Pārlūkot izejas kodu

Simplify and fix `--save-period` epoch 0 (#8042)

Simplify and update `--save-period` epoch 0
modifyDataloader
Glenn Jocher GitHub pirms 2 gadiem
vecāks
revīzija
dcf8073881
Šim parakstam datu bāzē netika atrasta zināma atslēga GPG atslēgas ID: 4AEE18F83AFDEB23
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. +1
    -1
      train.py

+ 1
- 1
train.py Parādīt failu

@@ -423,7 +423,7 @@ def train(hyp, opt, device, callbacks): # hyp is path/to/hyp.yaml or hyp dictio
torch.save(ckpt, last)
if best_fitness == fi:
torch.save(ckpt, best)
if (epoch > 0) and (opt.save_period > 0) and (epoch % opt.save_period == 0):
if opt.save_period > 0 and epoch % opt.save_period == 0:
torch.save(ckpt, w / f'epoch{epoch}.pt')
del ckpt
callbacks.run('on_model_save', last, epoch, final_epoch, best_fitness, fi)

Notiek ielāde…
Atcelt
Saglabāt