Remove sync_dist from class metrics as they are automatically reduced

Lightning-Universe · Nov 16, 2020 · df0c174 · df0c174
1 parent 0d08e6d
commit df0c174
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pl_bolts/models/self_supervised/ssl_finetuner.py b/pl_bolts/models/self_supervised/ssl_finetuner.py
@@ -106,7 +106,7 @@ def validation_step(self, batch, batch_idx):
         acc = self.val_acc(logits, y)
 
         self.log('val_loss', loss, prog_bar=True, sync_dist=True)
-        self.log('val_acc', self.val_acc, sync_dist=True)
+        self.log('val_acc', self.val_acc)
 
         return loss
 
@@ -115,7 +115,7 @@ def test_step(self, batch, batch_idx):
         acc = self.test_acc(logits, y)
 
         self.log('test_loss', loss, sync_dist=True)
-        self.log('test_acc', self.test_acc, sync_dist=True)
+        self.log('test_acc', self.test_acc)
 
         return loss