- Linux
- Anaconda 3
- Python 3.8
- CUDA 11.1
- RTX 3090
To get started, please create the conda environment pvcs
by running
conda env create -f environment.yml
Please also install Pytorch3D
To train the model by running
bash train.sh
To train the model by running
bash test.sh