Issues with the env

[tenres]

Hi, thank you for making this code available. I am trying to get it to run on the specified env, but have been getting some errors. Initially, I got the "module was missing Resize attribute" error, which required updating torchvision. Then I got the "cuDNN error: CUDNN_STATUS_INTERNAL_ERROR". I updated pytorch to 1.1 and cudnn but it didn't help. When I set cudnn.benchmark = False, I don't get the error anymore. Have you seen these before in your env? By any chance, do you happen to have a docker available with your env? Any suggestions will be helpful. Thanks.