(32, 32) -> (128, 128)

odulcy-mindee · Apr 16, 2024 · bd18864 · bd18864
1 parent 05a6779
commit bd18864
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/references/classification/train_pytorch_orientation.py b/references/classification/train_pytorch_orientation.py
@@ -236,7 +236,7 @@ def main(args):
 
     torch.backends.cudnn.benchmark = True
 
-    input_size = (256, 256) if args.type == "page" else (32, 32)
+    input_size = (256, 256) if args.type == "page" else (128, 128)
 
     # Load val data generator
     st = time.time()