enable semantic loss by default

OpenGVLab · Mar 26, 2024 · 0c5af29 · 0c5af29
1 parent b890d36
commit 0c5af29
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/scannet/pretrain-ponder-ppt-v1m1-0-sc-s3-st-spunet.py b/configs/scannet/pretrain-ponder-ppt-v1m1-0-sc-s3-st-spunet.py
@@ -206,7 +206,7 @@
         (0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13, 15, 20, 22, 24, 25, 27, 34),
         (0, 1, 4, 5, 6, 7, 8, 10, 19, 29, 30, 31, 32),
     ),
-    ppt_loss_weight=0.0,
+    ppt_loss_weight=1.0,
     ppt_criteria=[dict(type="CrossEntropyLoss", loss_weight=1.0, ignore_index=-1)],
 )
 

diff --git a/ponder/models/ponder/ponder_indoor_base.py b/ponder/models/ponder/ponder_indoor_base.py
@@ -38,7 +38,7 @@ def __init__(
         clip_model=None,
         class_name=None,
         valid_index=None,
-        ppt_loss_weight=0.0,  # whether and how much to use PPT's loss
+        ppt_loss_weight=1.0,  # whether and how much to use PPT's loss
         ppt_criteria=None,
     ):
         super().__init__()