search

allenai / procthor-rl

RL training scripts for learning an agent using ProcTHOR.
Apache License 2.0
11 stars 3 forks source link

How much resource required? #6

Open NinjaCXX opened 1 month ago

NinjaCXX commented 1 month ago

Thanks for your great work. I want to follow Embodied-Codebook. How many GPU resources do you use and how long does the training process last roughly?

I use a 3090 GPU(24G), and I find it will lead to GPU out of memory if I maintain the machine.num_train_processes as 96. I manage the start training with machine.num_train_processes of 8. I worry that I can not continue my research if it indeed needs too many GPU resources. @KuoHaoZeng @Ainaz99 Thanks for your reply~

NinjaCXX commented 1 month ago

Can you give me the approximate time?