HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix Multi-GPU resume bug (#7328)

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Adamcode 2024-01-05 23:49:17 +08:00 • committed by GitHub
parent a334ff7f6e
commit eeea5de623
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
ultralytics/engine/trainer.py
View file

@ -121,7 +121,6 @@ class BaseTrainer:
self.trainset, self.testset = self.get_dataset(self.data) self.trainset, self.testset = self.get_dataset(self.data)
self.ema = None self.ema = None
self.resume = False
# Optimization utils init # Optimization utils init
self.lf = None self.lf = None