updates
This commit is contained in:
parent
70ba7f0805
commit
ea11af6132
2
train.py
2
train.py
|
@ -287,7 +287,7 @@ def train(
|
||||||
del chkpt
|
del chkpt
|
||||||
|
|
||||||
dt = (time.time() - t0) / 3600
|
dt = (time.time() - t0) / 3600
|
||||||
print('%g epochs completed in %.3f hours.' % (epoch - start_epoch, dt))
|
print('%g epochs completed in %.3f hours.' % (epoch - start_epoch + 1, dt))
|
||||||
return results
|
return results
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue