search

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

the beginnings of support for PyTorch v0.4.0 #21

Closed aneeshchauhan closed 5 years ago

aneeshchauhan commented 5 years ago

Changes to model.py and trainer.py model.py: Fixed bug on usage of switching gradient computation on/off trainer.py: Fixed bug on handling 0 dimensional loss_values. This is an error I faced while running the tests