updates
This commit is contained in:
parent
ef84864251
commit
7f2df90277
6
train.py
6
train.py
|
@ -157,9 +157,9 @@ def main(opt):
|
||||||
# if i == 1:
|
# if i == 1:
|
||||||
# return
|
# return
|
||||||
|
|
||||||
# Write epoch results
|
# Write epoch results
|
||||||
with open('results.txt', 'a') as file:
|
with open('results.txt', 'a') as file:
|
||||||
file.write(s + '\n')
|
file.write(s + '\n')
|
||||||
|
|
||||||
# Update best loss
|
# Update best loss
|
||||||
loss_per_target = rloss['loss'] / rloss['nGT']
|
loss_per_target = rloss['loss'] / rloss['nGT']
|
||||||
|
|
Loading…
Reference in New Issue