Hi, it shows me the following error, seems the normalize has some issue
File "test_xy_final.py", line 62, in <module>
    main()
  File "test_xy_final.py", line 43, in main
    img_tensor = compose_img(img_tensor)
  File "/opt/conda/lib/python3.6/site-packages/torchvision/transforms/transforms.py", line 61, in __call__
    img = t(img)
  File "/opt/conda/lib/python3.6/site-packages/torchvision/transforms/transforms.py", line 166, in __call__
    return F.normalize(tensor, self.mean, self.std, self.inplace)
  File "/opt/conda/lib/python3.6/site-packages/torchvision/transforms/functional.py", line 217, in normalize
    tensor.sub_(mean[:, None, None]).div_(std[:, None, None])
RuntimeError: output with shape [1, 112, 112] doesn't match the broadcast shape [3, 112, 112]