dict typo

Thomas Capelle · Thomas Capelle · commit 87c80f0ab325 · 2021-11-25T20:58:01.000+01:00
diff --git a/scripts/keras_cvp.py b/scripts/keras_cvp.py
@@ -57,7 +57,7 @@ def on_epoch_end(self, epoch, logs={}):
         self.batch_times.sort()
         avg_time_per_batch = sum(self.batch_times[0:-self.drop])/(len(self.batch_times)-self.drop)
         samples_s_batch = self.batch_size / avg_time_per_batch
-        wandb.log({"samples_per_batch", samples_s_batch}, step=epoch)
+        wandb.log({"samples_per_batch": samples_s_batch}, step=epoch)
         self.samples_s += samples_s_batch
     
     def on_train_end(self, logs={}):
