minor changes

train_tongue_diagnosis.py

@@ -16,7 +16,7 @@ parser.add_argument('--model_name', default='resnet-50',
                          'resnet-152')
 parser.add_argument('--gpuid', default='0', type=str, help='which gpu to use')
 parser.add_argument('--no_cuda', action='store_true', default=False, help='disables CUDA training')
-parser.add_argument('-b', '--batch_size', default=2, type=int, help='mini-batch size (default: 64)')
+parser.add_argument('-b', '--batch_size', default=32, type=int, help='mini-batch size (default: 32)')
 parser.add_argument('--lr', type=float, default=0.1, metavar='LR', help='learning rate')
 parser.add_argument('--momentum', type=float, default=0.9, metavar='M', help='SGD momentum')
 parser.add_argument('--weight_decay', '--wd', default=2e-4, type=float, metavar='W')