an unofficial implementation of robotap
We use the robosuite environment due to my limit desktop gpu. If you find some environments that don't need lots of computing resources, don't hesitate to tell me.
First, you should prepare for the robosuite environment:
conda create -n tapnet python=3.11
pip install mujoco
pip install robosuite
Then, prepare for tapnet methods according to the official repository:
git clone https://github.com/deepmind/tapnet.git
cd tapnet
pip install .
Then, download the pre-trained model:
mkdir checkpoints
wget -P checkpoints https://storage.googleapis.com/dm-tapnet/tapir_checkpoint_panning.npy
wget -P checkpoints https://storage.googleapis.com/dm-tapnet/causal_tapir_checkpoint.npy