Go2 distillation requires 30 days?
CoderWangcai opened this issue · comments
I am using a 4090 graphics card with 24GB of memory. When running python legged_gym/scripts/train.py --headless --task go2_distill, I encounter a CUDA out-of-memory error, as shown in the image below:
Therefore, I changed num_envs = 256 to num_envs = 128 in the go2_distill_config.py, but I found that it would take over 30 days to complete the training, as shown in the image below. Is this normal?
I recommend using multi-process distillation.