From c36e09d5955d3bfe4c06ea14645357f574fc5453 Mon Sep 17 00:00:00 2001
From: yuedongli1 <yuedongli1@huawei.com>
Date: Fri, 20 Sep 2024 10:00:10 +0800
Subject: [PATCH] add scale for loss hyps

---
 train.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/train.py b/train.py
index cecce995..a02196bd 100644
--- 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