更新 'gpu/train.py'

gpu/train.py

@@ -37,7 +37,7 @@ parser = argparse.ArgumentParser(description='PyTorch MNIST Example')
 parser.add_argument('--traindata', default="/dataset/train" ,help='path to train dataset')
 parser.add_argument('--testdata', default="/dataset/test" ,help='path to test dataset')
 parser.add_argument('--epoch_size', type=int, default=1, help='how much epoch to train')
-parser.add_argument('--batch_size', type=int, default=256, help='how much batch_size in epoch')
+parser.add_argument('--batch_size', type=int, default=20, help='how much batch_size in epoch')
 
 if __name__ == '__main__':
     args, unknown = parser.parse_known_args()