xiaoye-hhh / SAAI

17 stars 1 forks source link

code #9

Open Velpro-collab opened 5 months ago

Velpro-collab commented 5 months ago

logits_m = torch.cat([logits_v, logits_i], 0).float() with torch.no_grad(): self.infraredclassifier.weight.data = self.infraredclassifier.weight.data * (1 - self.update_rate) \