minor change

train_tongue_diagnosis.py

@@ -16,15 +16,15 @@ parser = argparse.ArgumentParser(description='PyTorch DenseNet Training')
 parser.add_argument('--model_name', default='resnet-50',
                     help='models: densenet121, resnet-50, resnet-101, inception_v, vgg19_bn')
 parser.add_argument('--gpuid', default='0', type=str, help='which gpu to use')
-parser.add_argument('--no_cuda', action='store_true', default=True, help='disables CUDA training')
-parser.add_argument('-b', '--batch_size', default=2, 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('--no_cuda', action='store_true', default=False, help='disables CUDA training')
+parser.add_argument('-b', '--batch_size', default=32, type=int, help='mini-batch size (default: 32)')
+parser.add_argument('--lr', type=float, default=0.05, 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')
-parser.add_argument('--epochs', type=int, default=200, metavar='N', help='number of epochs to train')
-parser.add_argument('--schedule', type=int, nargs='+', default=[100, 150],
+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],
                     help='Decrease learning rate at these epochs.')
-parser.add_argument('--gamma', type=float, default=0.1, help='LR is multiplied by gamma on schedule.')
+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')
 parser.add_argument('--save_freq', '-s', default=1, type=int, metavar='N', help='save frequency')
 parser.add_argument('--log_interval', type=int, default=10, metavar='N',
@@ -56,7 +56,7 @@ transform_train = transforms.Compose([
     # transforms.RandomCrop((224, 224), padding=14),
     transforms.RandomHorizontalFlip(),
     transforms.ToTensor(),
-    # normalizer
+    normalizer
 ])
 
 tongue_dataset_train = Image_DataSet(transform=transform_train)