Support pytorch v0.4 - Githubissues

andyzeng / visual-pushing-grasping

Train robotic agents to learn to plan pushing and grasping actions for manipulation with deep reinforcement learning.

http://vpg.cs.princeton.edu/

BSD 2-Clause "Simplified" License

898 stars 315 forks source link

Closed aneeshchauhan closed 5 years ago

aneeshchauhan commented 5 years ago

bug removed in gradient switching using pytorch 0.4 (models.py); handling 0dim arrays in trainer.py

andyzeng commented 5 years ago

This is great, thanks Aneesh!