diff --git a/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml b/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml index c78c72df1952aee1064cd14a9fe97a65defa3475..52b2ec145a39b8093c5fe521074ce4e611c32de9 100644 --- a/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml @@ -31,7 +31,7 @@ OPTIMIZER: factor: 0.0001 TRAIN: - batch_size: 64 + batch_size: 256 num_workers: 4 file_list: "./dataset/ILSVRC2012/train_list.txt" data_dir: "./dataset/ILSVRC2012/"