minor changes

train_tongue_diagnosis.py

@@ -21,8 +21,8 @@ parser.add_argument('-b', '--batch_size', default=32, type=int, help='mini-batch
 parser.add_argument('--lr', type=float, default=0.001, 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=0, type=float, metavar='W')
-parser.add_argument('--epochs', type=int, default=20, metavar='N', help='number of epochs to train')
-parser.add_argument('--schedule', type=int, nargs='+', default=[10, 15],
+parser.add_argument('--epochs', type=int, default=40, metavar='N', help='number of epochs to train')
+parser.add_argument('--schedule', type=int, nargs='+', default=[25, 30, 35],
                     help='Decrease learning rate at these epochs.')
 parser.add_argument('--gamma', type=float, default=0.2, help='LR is multiplied by gamma on schedule.')
 parser.add_argument('--model_dir', default='dnn_models/tongue_modes/', help='directory of model for saving checkpoint')