nikhilbarhate99 / PPO-PyTorch

Minimal implementation of clipped objective Proximal Policy Optimization (PPO) in PyTorch
MIT License
1.66k stars 343 forks source link

Setting Model to eval() mode in test.py #57

Open rllyryan opened 2 years ago

rllyryan commented 2 years ago

Dear @nikhilbarhate99,

Can I ask if we need to set the models to evaluation mode (nn.Module.eval()) in the test script to see if the weights are suitable?

Thank you!