release commit
authorManu Mathew <a0393608@ti.com>
Mon, 27 Jan 2020 07:27:13 +0000 (12:57 +0530)
committerManu Mathew <a0393608@ti.com>
Mon, 27 Jan 2020 07:27:13 +0000 (12:57 +0530)
modules/pytorch_jacinto_ai/engine/train_pixel2pixel.py

index c9b0285ca3ca11d642d802a0079246d22c9e2023..25d2f42827ef5dd1b9c27503e06e332eb6c0dc7b 100644 (file)
@@ -1027,7 +1027,7 @@ def get_train_transform(args):
 
     # crop size used only for training
     image_train_output_scaling = vision.transforms.image_transforms.Scale(args.rand_resize, target_size=args.rand_output_size, is_flow=args.is_flow) \
-        if (args.rand_output_size and args.rand_output_size != args.rand_resize) else None
+        if (args.rand_output_size is not None and args.rand_output_size != args.rand_resize) else None
     train_transform = vision.transforms.image_transforms.Compose([
         image_prenorm,
         vision.transforms.image_transforms.AlignImages(),