Kaynağa Gözat

Merge remote-tracking branch 'origin/master'

5.0
Glenn Jocher 4 yıl önce
ebeveyn
işleme
2b7a11300e
1 değiştirilmiş dosya ile 2 ekleme ve 0 silme
  1. +2
    -0
      train.py

+ 2
- 0
train.py Dosyayı Görüntüle

@@ -134,6 +134,8 @@ def train(hyp):
file.write(ckpt['training_results']) # write results.txt

start_epoch = ckpt['epoch'] + 1
assert opt.epochs > start_epoch, '%s has already trained %g epochs. --epochs must be greater than %g' % \
(opt.weights, ckpt['epoch'], ckpt['epoch'])
del ckpt

# Mixed precision training https://github.com/NVIDIA/apex

Yükleniyor…
İptal
Kaydet