Merge branch 'fix-teardown-barrier' of github.com:ananthsub/pytorch-l…

…ightning into fix-teardown-barrier
Lightning-AI · Apr 13, 2021 · 9239ee3 · 9239ee3
2 parents 8b900b2 + 5d83193
commit 9239ee3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -1931,11 +1931,11 @@ class TestModel(BoringModel):
         def validation_step(self, batch, batch_idx):
             output = self.layer(batch)
             loss = self.loss(batch, output)
-            self.log('avg_x', loss)
+            self.log('x', loss)
 
 
     model = TestModel()
-    checkpoint = ModelCheckpoint(dirpath=tmpdir, monitor='avg_x', mode='min', save_top_k=1)
+    checkpoint = ModelCheckpoint(dirpath=tmpdir, monitor='x', mode='min', save_top_k=1)
     trainer = Trainer(
         default_root_dir=tmpdir,
         max_epochs=2,