minor bug fix in eval (in transforms.Resize)

2020-05-22 16:00:59 +05:30 · 2020-05-22 16:00:59 +05:30 · 5af6d8b71e
parent 9198faec92
commit 5af6d8b71e
1 changed files with 1 additions and 1 deletions
--- a/projects/attribute_recognition/main.py
+++ b/projects/attribute_recognition/main.py
@ -42,7 +42,7 @@ def init_dataset(use_gpu):
    )

    transform_te = Compose(
-        [Resize(args.height, args.width),
+        [Resize([args.height, args.width]),
         ToTensor(), normalize]
    )