RuntimeError: CUDA out of memory

xukechun / Efficient_goal-oriented_push-grasping_synergy

[RAL & IROS 2021] Efficient learning of goal-oriented push-grasping synergy in clutter

60 stars 10 forks source link

RuntimeError: CUDA out of memory #11

Open andreipit opened 2 years ago

andreipit commented 2 years ago

Thank you guys, for a very interesting project! Got "RuntimeError: CUDA out of memory". Training with --cpu is very slow. Is there any other way to train on the old gpu? How to reduce the batch size?

duweiwmxes commented 1 year ago

I'm having the same issue, can you tell me how to fix it.