mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 21:44:22 +08:00
Fix Multi-GPU resume bug (#7328)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
a334ff7f6e
commit
eeea5de623
@ -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
|
||||||
|
Loading…
x
Reference in New Issue
Block a user