Install torch and torchvision according to your environment. For reference, we use torch==2.0.1+cu118. Install other packages. NaturalTransRecon/ |-- data/ |-- case ...