Error when loading viz.py

[RaymonHE]

Hi Davis, I created a docker image so that I can build the same environment as you mentioned here in the GitHub.
When I was about to run this command,
python viz.py --data-cfg ../data/configs/demo.cfg --weights ../pretrained_weights/caspr_weights_cars.pth --seq-len 10 --num-pts 2048 --viz-tnocs --viz-observed --viz-interpolated

The output in the terminal is

[RaymonHE]

By the way, Davis, I've successfully created the docker image file which satisfies all the environment requirements. I can share that.