diff --git a/birdwatch/callbacks/trainingmonitor.py b/birdwatch/callbacks/trainingmonitor.py index e9cc156..6cfc462 100644 --- a/birdwatch/callbacks/trainingmonitor.py +++ b/birdwatch/callbacks/trainingmonitor.py @@ -36,7 +36,7 @@ def on_epoch_end(self, epoch, logs={}): # for the entire training process for (k, v) in logs.items(): l = self.H.get(k, []) - l.append(v.item()) + l.append(v) self.H[k] = l # check to see if the training history should be serialized @@ -84,4 +84,4 @@ def on_epoch_end(self, epoch, logs={}): # save the figure plt.savefig(self.figPath) - plt.close() \ No newline at end of file + plt.close()