Solved by driazati in post #2 
              
              
                You may be on an old version of torchvision or torch, can you make sure you’re on the latest of both? (or use the nightly if there are still errors) 
Your code works fine for me after applying the following diff: 
diff --git a/test.py b/test.py
index e305823abd..a96eb93829 100644
--- a/test.py
+++ b…