set training batch size for classification

Author: taigw

classification/AntBee/config/train_test_ce1.cfg

@@ -8,6 +8,8 @@ train_csv = config/train_data.csv
 valid_csv = config/valid_data.csv
 test_csv  = config/valid_data.csv
 
+train_batch_size = 4
+
 # modality number
 modal_num = 1
 
@@ -59,7 +61,6 @@ class_num = 2
 # list of gpus
 gpus       = [0]
 
-batch_size    = 4
 loss_type     = CrossEntropyLoss
 
 # for optimizers
@@ -84,7 +85,6 @@ iter_save  = 500
 [testing]
 # list of gpus
 gpus        = [0]
-batch_size  = 4
 
 # checkpoint mode can be [0-latest, 1-best, 2-specified]
 ckpt_mode        = 1

classification/AntBee/config/train_test_ce2.cfg

@@ -8,6 +8,8 @@ train_csv = config/train_data.csv
 valid_csv = config/valid_data.csv
 test_csv  = config/valid_data.csv
 
+train_batch_size = 4
+
 # modality number
 modal_num = 1
 
@@ -60,7 +62,6 @@ class_num = 2
 # list of gpus
 gpus       = [0]
 
-batch_size    = 4
 loss_type     = CrossEntropyLoss
 
 # for optimizers
@@ -85,7 +86,6 @@ iter_save  = 500
 [testing]
 # list of gpus
 gpus        = [0]
-batch_size  = 4
 
 # checkpoint mode can be [0-latest, 1-best, 2-specified]
 ckpt_mode        = 1

classification/CHNCXR/config/net_resnet18.cfg

@@ -8,6 +8,8 @@ train_csv = config/cxr_train.csv
 valid_csv = config/cxr_valid.csv
 test_csv  = config/cxr_test.csv
 
+train_batch_size = 4
+
 # modality number
 modal_num = 1
 
@@ -60,7 +62,6 @@ class_num = 2
 # list of gpus
 gpus       = [0]
 
-batch_size    = 4
 loss_type     = CrossEntropyLoss
 
 # for optimizers
@@ -85,7 +86,6 @@ iter_save  = 1000
 [testing]
 # list of gpus
 gpus        = [0]
-batch_size  = 4
 
 # checkpoint mode can be [0-latest, 1-best, 2-specified]
 ckpt_mode        = 1

classification/CHNCXR/config/net_vgg16.cfg

@@ -8,6 +8,8 @@ train_csv = config/cxr_train.csv
 valid_csv = config/cxr_valid.csv
 test_csv  = config/cxr_test.csv
 
+train_batch_size = 4
+
 # modality number
 modal_num = 1
 
@@ -61,7 +63,6 @@ class_num = 2
 # list of gpus
 gpus       = [0]
 
-batch_size    = 4
 loss_type     = CrossEntropyLoss
 
 # for optimizers
@@ -86,7 +87,6 @@ iter_save  = 1000
 [testing]
 # list of gpus
 gpus        = [0]
-batch_size  = 4
 
 # checkpoint mode can be [0-latest, 1-best, 2-specified]
 ckpt_mode        = 1