Merge pull request #357 from yuedongli1/master

add scale for loss hyps
mindspore-lab · Sep 20, 2024 · 3834320 · 3834320
2 parents 1868066 + c36e09d
commit 3834320
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/train.py b/train.py
@@ -188,6 +188,10 @@ def train(args):
     else:
         eval_dataset, eval_dataloader = None, None
 
+    # Scale loss hyps
+    args.loss.cls *= args.data.nc / 80
+    args.loss.obj *= (args.img_size / 640) ** 2
+
     # Create Loss
     loss_fn = create_loss(
         **args.loss, anchors=args.network.get("anchors", 1), stride=args.network.stride, nc=args.data.nc