Li_Ting_Tay:

torchvision package is v0.1.12.


Likely, the torchvision package is still not quite up to date (0.1.12 sounds like a torch version). So you may need the source:

  
      
      GitHub
  
  
    

pytorch/vision 117

Datasets, Transforms and Models specific to Computer Vision - pytorch/vision


  
  
    
    
  
  


Best regards
Thomas