Jacob Rafati

icon indicating a website link http://rafati.net icon indicating an email [email protected]

icon company Workout Vision icon location Seattle, WA icon location Ph.D. of Computer Science. Machine Learning Research Scientist.

Results 2 comments of Jacob Rafati

No module named gym

I am using ipython, and I had the same problem. I noticed that ipython didn't belong to the Anaconda pkg. So I installed ipython for my anaconda environment `pip install...

Key mismatch while loading the model?

If you want to load the weights after `DataParallel` use: `net.module.load_state_dict(pertained_weights)` If you want to load the weights before `DataParallel` use: `net.load_state_dict(pertained_weights)`