From aec12fa510dd0ba7b87326a61525b66a1b881e0f Mon Sep 17 00:00:00 2001 From: unknown <2660417142@qq.com> Date: Wed, 24 Sep 2025 16:27:51 +0800 Subject: [PATCH] cancel scale loss hyps --- train.py | 9 --------- 1 file changed, 9 deletions(-) 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(