Code for Nonparametric Tensor Ring Decomposition with Scalable Amortized Inference

Description

The code is mainly based on PyTorch. Detailed information is listed in ./requirements.txt

• For baseline methods, run python baseline_indoor.py or python baseline_radar.py.
• For HI-VAE, run bash run_HI-VAE.sh
• For GP-VAE, run bash run_GP-VAE.sh
• For our method, run bash run_NTRD.sh