batch size 16

liuzhengzhe · Aug 18, 2021 · 07769ea · 07769ea
1 parent d5f92e1
commit 07769ea
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/3d-2d-distillation/config/scannet/scannet_pspnet50.yaml b/3d-2d-distillation/config/scannet/scannet_pspnet50.yaml
@@ -17,9 +17,9 @@ TRAIN:
   zoom_factor: 8  # zoom factor for final prediction during training, be in [1, 2, 4, 8]
   ignore_label: 255
   aux_weight: 0.4
-  train_gpu: [0]
-  workers: 2  # data loader workers
-  batch_size: 2  # batch size for training
+  train_gpu: [0,1,2,3]
+  workers: 16  # data loader workers
+  batch_size: 16  # batch size for training
   batch_size_val: 8  # batch size for validation during training, memory and speed tradeoff
   base_lr: 0.01
   epochs: 50