fixed undesired behaviour due to dict.fromkeys

Lightning-AI · May 25, 2020 · 21b14a7 · 21b14a7
1 parent 8ca8336
commit 21b14a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/callbacks/lr_logger.py b/pytorch_lightning/callbacks/lr_logger.py
@@ -59,7 +59,7 @@ def on_train_start(self, trainer, pl_module):
         names = self._find_names(trainer.lr_schedulers)
 
         # Initialize for storing values
-        self.lrs = dict.fromkeys(names, [])
+        self.lrs = {name: [] for name in names}
 
     def on_batch_start(self, trainer, pl_module):
         latest_stat = self._extract_lr(trainer, 'step')