ultralytics 8.0.86
HUB resume, Classify train and RayTune fixes (#2200)
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
This commit is contained in:
@ -30,7 +30,7 @@ class ClassificationTrainer(BaseTrainer):
|
||||
if weights:
|
||||
model.load(weights)
|
||||
|
||||
pretrained = False
|
||||
pretrained = self.args.pretrained
|
||||
for m in model.modules():
|
||||
if not pretrained and hasattr(m, 'reset_parameters'):
|
||||
m.reset_parameters()
|
||||
|
Reference in New Issue
Block a user