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 index ded156f..9ae649e 100644 --- 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 index daff7fb..4f6749a 100644 --- 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__()