fix loss hyps

mindspore-lab · Oct 17, 2024 · 834489e · 834489e
1 parent 13c0f0d
commit 834489e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/train.py b/train.py
@@ -189,11 +189,9 @@ def train(args):
         eval_dataset, eval_dataloader = None, None
 
     # Scale loss hyps
-    nl = len(args.network.anchors)
-    args.loss.box *= 3 / nl  # scale to layers
-    args.loss.cls *= args.data.nc / 80 * 3 / nl  # scale to classes and layers
+    args.loss.cls *= args.data.nc / 80  # scale to classes and layers
     if hasattr(args.loss, "obj"):
-        args.loss.obj *= (args.img_size / 640) ** 2 * 3 / nl  # scale to image size and layers
+        args.loss.obj *= (args.img_size / 640) ** 2  # scale to image size and layers
 
     # Create Loss
     loss_fn = create_loss(