Skip to content

Commit cc52487

Browse files
authored
Update callbacks.py
1 parent 33ecf5c commit cc52487

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils/callbacks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def __init__(self, log_dir):
1616

1717
os.makedirs(self.save_path)
1818

19-
def append_loss(self, loss, val_loss):
19+
def append_loss(self, loss):
2020
self.losses.append(loss)
2121
with open(os.path.join(self.save_path, "epoch_loss_" + str(self.time_str) + ".txt"), 'a') as f:
2222
f.write(str(loss))

0 commit comments

Comments
 (0)