updates
This commit is contained in:
parent
54e43b9ad6
commit
a34a760d0f
2
train.py
2
train.py
|
@ -188,7 +188,7 @@ def train(
|
||||||
# Update best loss
|
# Update best loss
|
||||||
test_loss = results[4]
|
test_loss = results[4]
|
||||||
if test_loss < best_loss:
|
if test_loss < best_loss:
|
||||||
best_loss = results[0]
|
best_loss = test_loss
|
||||||
|
|
||||||
# Save training results
|
# Save training results
|
||||||
save = True and not opt.nosave
|
save = True and not opt.nosave
|
||||||
|
|
Loading…
Reference in New Issue