Seg objective only for unique path ultralytics#6

manole-alexandru · Mar 24, 2023 · 620f1c3 · 620f1c3
1 parent 72d83c4
commit 620f1c3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/train.py b/train.py
@@ -511,7 +511,7 @@ def parse_opt(known=False):
 
 
 def main(opt, callbacks=Callbacks()):
-    print('\n---------- VERSION:', '#0005', '----------\n')
+    print('\n---------- VERSION:', '#0006', '----------\n')
     # Checks
     if RANK in {-1, 0}:
         print_args(vars(opt))

diff --git a/utils/torch_utils.py b/utils/torch_utils.py
@@ -355,8 +355,8 @@ def smart_optimizer(model, name='Adam', lr=0.001, momentum=0.9, decay=1e-5):
                     g_det[1].append(p)
                 else:
                     g_det[0].append(p)  # weight (with decay)
-
-        if m_index < 287 or m_index > 290: # SEGMENTATION Optimizer
+        else:
+        # if m_index < 287 or m_index > 290: # SEGMENTATION Optimizer
             for p_index, (p_name, p) in enumerate(v.named_parameters(recurse=0)):
                 if p_name == 'bias':  # bias (no decay)
                     g_seg[2].append(p)
@@ -366,7 +366,7 @@ def smart_optimizer(model, name='Adam', lr=0.001, momentum=0.9, decay=1e-5):
                     g_seg[0].append(p)  # weight (with decay)
 
     det_optimizer = get_optimizer(name, lr, momentum, decay, g_det)
-    seg_optimizer = get_optimizer(name, lr, momentum, decay, g_seg)
+    seg_optimizer = get_optimizer(name, lr * 5, momentum, decay, g_seg)
     return det_optimizer, seg_optimizer