Solved by ptrblck in post #2 
              
              
                The code works using torchvision==0.8.0.dev20201006 and it seems the functionality was added in this PR. 
You could update to the latest nightly binary, build from source, or wait for the next stable release.