diff --git a/live_loss_plot.py b/live_loss_plot.py index 406a067..1756169 100644 --- a/live_loss_plot.py +++ b/live_loss_plot.py @@ -24,7 +24,7 @@ def on_train_begin(self, logs={}): self.logs = [] def on_epoch_end(self, epoch, logs={}): - self.logs.append(logs) + self.logs.append(logs.copy()) clear_output(wait=True) plt.figure(figsize=self.figsize)