back rename

tcapelle · tcapelle · commit a8e6462b1a6b · 2023-06-21T15:57:48.000+02:00
diff --git a/pytorch/utils.py b/pytorch/utils.py
@@ -90,7 +90,7 @@ def do_one_epoch(self, dl, epoch):
             metrics = {"train/train_loss": loss, 
                         "train/epoch": (step + 1 + (self.n_steps_per_epoch * epoch)) / self.n_steps_per_epoch, 
                         "train/example_ct": self.example_ct,
-                        "seq_per_sec":len(batch["labels"])/(tf-ti),
+                        "sqe_per_sec":len(batch["labels"])/(tf-ti),
                         "seq_per_sec_dl":len(batch["labels"])/tf_with_dataloader,}
             if step + 1 < self.n_steps_per_epoch:
                 # 🐝 Log train metrics to wandb 
