fix dal-base config

HuangJunJie2017 · Nov 14, 2023 · ce387e5 · ce387e5
1 parent 867fcec
commit ce387e5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/configs/dal/dal-base.py b/configs/dal/dal-base.py
@@ -137,7 +137,7 @@
         instance_attn=False,
 
         # Transfusion
-        num_proposals=300,
+        num_proposals=200,
         in_channels=512,
         hidden_channel=128,
         num_classes=10,
@@ -267,7 +267,7 @@
         file_client_args=file_client_args),
     dict(
         type='LoadPointsFromMultiSweeps',
-        sweeps_num=9,
+        sweeps_num=10,
         use_dim=[0, 1, 2, 3, 4],
         file_client_args=file_client_args,
         pad_empty_sweeps=True,
@@ -306,7 +306,7 @@
         file_client_args=file_client_args),
     dict(
         type='LoadPointsFromMultiSweeps',
-        sweeps_num=9,
+        sweeps_num=10,
         use_dim=[0, 1, 2, 3, 4],
         file_client_args=file_client_args,
         pad_empty_sweeps=True,
@@ -379,3 +379,4 @@
 optimizer = dict(type='AdamW', lr=2e-4, weight_decay=0.01)  # for 64 total batch size
 two_stage = True
 runner = dict(type='TwoStageRunner', max_epochs=20)
+num_proposals_test = 300