diff --git a/train.py b/train.py index a68dadf6..9472412e 100644 --- a/train.py +++ b/train.py @@ -189,15 +189,6 @@ def train(args): ) else: eval_dataset, eval_dataloader = None, None - - # Scale loss hyps - nl = network.model.model[-1].nl - if hasattr(args.loss, "box"): - args.loss.box *= 3 / nl # scale to layers - if hasattr(args.loss, "cls"): - args.loss.cls *= args.data.nc / 80 * 3 / nl # scale to classes and layers - if args.anchor_base and hasattr(args.loss, "obj"): - args.loss.obj *= (args.img_size / 640) ** 2 * 3 / nl # scale to image size and layers # Create Loss loss_fn = create_loss(