Commit 447ca5ee authored by Jonas Müller's avatar Jonas Müller

better output

parent 9bcf8e78
......@@ -210,7 +210,8 @@ class_mapping_inv = {v: k for k, v in class_mapping.items()}
print('Starting training')
vis = True
log = open(options.out_log, "w")
log = open(options.out_log, "a")
log.write("NEW RUN")
for epoch_num in range(num_epochs):
progbar = None
......@@ -312,7 +313,8 @@ for epoch_num in range(num_epochs):
print(out)
log.write("---\nEpoch "+str(epoch_num)+"\n")
log.write(out + "\n")
log.flush()
best_loss = curr_loss
best_loss_epoch_num = epoch_num
......
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