Sorry if this is a stupid question with an obvious answer but I’ve been messing around the example on github to train imagenet. Instead, I used it on my own custom dataset which is structured similarly. It seemed to train OK (could be better), but I don’t know how make the script run in evaluate mode. I know you use the -e flag for evaluate, but how do I tell it to use my newly created PTH file? I guess I could try writing my own code, but since the github example already has an eval mode…
Here’s the github code:

  
      
      GitHub
  
  
    

pytorch/examples

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