This example is very simple, but it works well. If you want more information. Check this Repository.

  
      
      GitHub
  
  
    

ShahriarSS/vision 40

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


  
  
    
    
  
  


There is also a way to not use JIT.
If it does not work, talk to me. I’ll tell you another way.