title: How to load a trained ML model with pytorch
body: <p>I am very new to Machine Learning, and very lack of the experience. Recently i am trying to collect my own datasets and see the performance of it on a <a href="https://github.com/Turoad/CLRNet" rel="nofollow noreferrer">trained model</a>. In README.md it only tells me how to train my own model, but it did not tell me how to load his model direct from <a href="https://github.com/turoad/CLRNet/releases" rel="nofollow noreferrer">here</a>.</p>
  <p>For instance, i want to try this trained model (culane_r18.pth.zip). How i am suppose to load this model? Should i start a new python file?</p>
  <pre><code># What should i add here?
  ... 
  
  FILE = &quot;culane_r18.pth&quot;
  
  device = torch.device('cpu')
  model = Model(*args, **kwargs)
  model.load_state_dict(torch.load(FILE, map_location=device))
  
  ...
  
  </code></pre>
  <p>I have only finished the <a href="https://www.youtube.com/watch?v=c36lUUr864M" rel="nofollow noreferrer">pytorch tutorial</a>, and now i do not know what should i do with is current model.</p>
  <p>Can anyone give me some help or tell me where should i find more details about this topic.</p>