Issue with running ppo

[szczekulskij]

Hi, I'm a CSE276F student. I'm trying to run ppo via 1quick_start notebook: https://colab.research.google.com/github/haosulab/ManiSkill/blob/main/examples/tutorials/1_quickstart.ipynb#scrollTo=Pm8hUvdYbWgH

When running below lines on Collab T4 machine:

!python ppo.py --env_id="PushCube-v1" --exp-name="state-pushcube" \
  --num_envs=1024 --update_epochs=8 --num_minibatches=32 \
  --total_timesteps=600_000 --eval_freq=8 --num-steps=20

I keep on getting the following error:

Could I request help please ?

[StoneT2000]

I have an idea of what the bug is (it might only appear in Colab and is caused by maybe gc.collect unloading modules) but this is actually fine. Your final model is saved still and the code ran to completion.