@Kalamaya have a look at examples when in doubt. the imagenet example and the dcgan example both have example uses:

  
      
      GitHub
  
  
    

pytorch/examples 337

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc. - pytorch/examples