Yes you can train see this book:

  
      
      pytorch.org
  
  
    

PyTorch 14

An open source deep learning platform that provides a seamless path from research prototyping to production deployment.


  
  
    
    
  
  


Picture11552×309 109 KB