Commit 2b59162e authored by Paul Bethge's avatar Paul Bethge
Browse files

add epoch to csv Callback

parent 8ca04555
......@@ -81,8 +81,10 @@ def write_csv(logging_dir, epoch, logs={}):
log_file_writer = csv.writer(log_file, delimiter=',')
if epoch == 0:
row = list(logs.keys())
row.insert(0, 'epoch')
log_file_writer.writerow(row)
row_vals = [round(x, 6) for x in list(logs.values())]
row_vals.insert(0, epoch)
log_file_writer.writerow(row_vals)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment