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={}): ...@@ -81,8 +81,10 @@ def write_csv(logging_dir, epoch, logs={}):
log_file_writer = csv.writer(log_file, delimiter=',') log_file_writer = csv.writer(log_file, delimiter=',')
if epoch == 0: if epoch == 0:
row = list(logs.keys()) row = list(logs.keys())
row.insert(0, 'epoch')
log_file_writer.writerow(row) log_file_writer.writerow(row)
row_vals = [round(x, 6) for x in list(logs.values())] row_vals = [round(x, 6) for x in list(logs.values())]
row_vals.insert(0, epoch)
log_file_writer.writerow(row_vals) 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