From d11f16c764b5d1e9454e847d7a0ac80df96183ad Mon Sep 17 00:00:00 2001 From: wjtest1201 Date: Mon, 15 Aug 2022 18:08:38 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'gpu/train.py'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gpu/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()