Merge pull request #383 from yuedongli1/bug_fix

fix yolox attribute error
mindspore-lab · Nov 12, 2024 · 31ba755 · 31ba755
2 parents 26f1b76 + 3b7cca6
commit 31ba755
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/train.py b/train.py
@@ -191,9 +191,11 @@ def train(args):
 
     # Scale loss hyps
     nl = network.model.model[-1].nl
-    args.loss.box *= 3 / nl  # scale to layers
-    args.loss.cls *= args.data.nc / 80 * 3 / nl  # scale to classes and layers
-    if args.anchor_base:
+    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