Release 8.0.4 fixes (#256)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: TechieG <35962141+gokulnath30@users.noreply.github.com> Co-authored-by: Parthiban Marimuthu <66585214+partheee@users.noreply.github.com>
This commit is contained in:
@ -143,6 +143,7 @@ def train(cfg):
|
||||
cfg.weight_decay = 5e-5
|
||||
cfg.label_smoothing = 0.1
|
||||
cfg.warmup_epochs = 0.0
|
||||
cfg.device = cfg.device if cfg.device is not None else ''
|
||||
# trainer = ClassificationTrainer(cfg)
|
||||
# trainer.train()
|
||||
from ultralytics import YOLO
|
||||
|
||||
Reference in New Issue
Block a user