diff --git a/gpu/train.py b/gpu/train.py
index ccedb05..6d82456 100755
--- a/gpu/train.py
+++ b/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()