diff --git a/train.py b/train.py index 56435dcc..a4af4af6 100644 --- a/train.py +++ b/train.py @@ -157,9 +157,9 @@ def main(opt): # if i == 1: # return - # Write epoch results - with open('results.txt', 'a') as file: - file.write(s + '\n') + # Write epoch results + with open('results.txt', 'a') as file: + file.write(s + '\n') # Update best loss loss_per_target = rloss['loss'] / rloss['nGT']