(self,x,fc)
| 154 | return new_fc |
| 155 | |
| 156 | def get_logits(self,x,fc): |
| 157 | if 'dot' in self.args.new_mode: |
| 158 | return F.linear(x,fc) |
| 159 | elif 'cos' in self.args.new_mode: |
| 160 | return self.args.temperature * F.linear(F.normalize(x, p=2, dim=-1), F.normalize(fc, p=2, dim=-1)) |
| 161 | |
| 162 | def update_fc_ft(self,new_fc,data,label,session): |
| 163 | new_fc=new_fc.clone().detach() |