We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 33ecf5c commit cc52487Copy full SHA for cc52487
utils/callbacks.py
@@ -16,7 +16,7 @@ def __init__(self, log_dir):
16
17
os.makedirs(self.save_path)
18
19
- def append_loss(self, loss, val_loss):
+ def append_loss(self, loss):
20
self.losses.append(loss)
21
with open(os.path.join(self.save_path, "epoch_loss_" + str(self.time_str) + ".txt"), 'a') as f:
22
f.write(str(loss))
0 commit comments