Update classifier.py

classifier.py

@@ -118,7 +118,7 @@ class Classifier:
                 grad.append(torch.as_tensor(grad_))
             # update weights
             grad = torch.stack(grad, dim=0)
-            self.weight -= 0.1 * grad
+            self.weight -= 0.01 * grad
             self.weight /= torch.norm(
                 self.weight, p=2, dim=-1, keepdim=True
             )