From 058d667dc953405f76e5c03b85cbd7c4eb70d128 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 9 Mar 2021 17:54:19 -0800 Subject: [PATCH] --img-size 64 --- classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/classifier.py b/classifier.py index 566de6cef2be..c8e8c0a856db 100644 --- a/classifier.py +++ b/classifier.py @@ -213,7 +213,7 @@ def test(model, dataloader, names, criterion=None, verbose=False, pbar=None): parser.add_argument('--hyp', type=str, default='data/hyp.scratch.yaml', help='hyperparameters path') parser.add_argument('--epochs', type=int, default=20) parser.add_argument('--batch-size', type=int, default=128, help='total batch size for all GPUs') - parser.add_argument('--img-size', type=int, default=32, help='train, test image sizes (pixels)') + parser.add_argument('--img-size', type=int, default=64, help='train, test image sizes (pixels)') parser.add_argument('--nosave', action='store_true', help='only save final checkpoint') parser.add_argument('--adam', action='store_true', help='use torch.optim.Adam() optimizer') parser.add_argument('--evolve', action='store_true', help='evolve hyperparameters')