add mini MNIST netcdf file example for testing

Author: KWang1998

examples/MNIST/MNIST_codes/main.py

@@ -132,10 +132,10 @@ def main():
     parser = argparse.ArgumentParser(description='PyTorch MNIST Example')
     parser.add_argument('--batch-size', type=int, default=64, metavar='N',
                         help='input batch size for training (default: 64)')
-    parser.add_argument('--test-batch-size', type=int, default=1000, metavar='N',
-                        help='input batch size for testing (default: 1000)')
-    parser.add_argument('--epochs', type=int, default=14, metavar='N',
-                        help='number of epochs to train (default: 14)')
+    parser.add_argument('--test-batch-size', type=int, default=2, metavar='N',
+                        help='input batch size for testing (default: 2)')
+    parser.add_argument('--epochs', type=int, default=3, metavar='N',
+                        help='number of epochs to train (default: 3)')
     parser.add_argument('--lr', type=float, default=1.0, metavar='LR',
                         help='learning rate (default: 1.0)')
     parser.add_argument('--gamma', type=float, default=0.7, metavar='M',
@@ -152,7 +152,7 @@ def main():
                         help='how many batches to wait before logging training status')
     parser.add_argument('--save-model', action='store_true', default=False,
                         help='For Saving the current Model')
-    parser.add_argument('--netcdf-file', type=str, default="../MNIST_data/mnist_images.nc",
+    parser.add_argument('--netcdf-file', type=str, default="../MNIST_data/mnist_images_mini.nc",
                         help='netcdf file storing train and test data')
     args = parser.parse_args()
     use_cuda = not args.no_cuda and torch.cuda.is_available()