From 07769ea6710b171573aa8a0190eb748f45a2cdca Mon Sep 17 00:00:00 2001 From: liuzhengzhe Date: Wed, 18 Aug 2021 16:04:45 +0800 Subject: [PATCH] batch size 16 --- 3d-2d-distillation/config/scannet/scannet_pspnet50.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/3d-2d-distillation/config/scannet/scannet_pspnet50.yaml b/3d-2d-distillation/config/scannet/scannet_pspnet50.yaml index 45265cc..c038609 100644 --- 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