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:
Glenn Jocher
2023-04-23 15:28:22 +02:00
committed by GitHub
parent 283e334bd4
commit 3d60347755
10 changed files with 52 additions and 26 deletions

View File

@ -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()