push

pytorch · Jul 25, 2023 · 799fd71 · 799fd71
1 parent 849ac96
commit 799fd71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchbenchmark/models/clip/__init__.py b/torchbenchmark/models/clip/__init__.py
@@ -40,7 +40,7 @@ def __init__(self, test, device, jit=False, batch_size=1, extra_args=[]):
         # Create optimizer
         self.loss_fn = ContrastiveLossWithTemperature()
         self.optimizer = torch.optim.AdamW(
-            list(self.model.parameters()) + list(loss_fn.parameters()),
+            list(self.model.parameters()) + list(self.loss_fn.parameters()),
             lr=5.0e-4,
             weight_decay=1.0e-4,
             eps=1.0e-6,