[CLIFF] segmentation fault when running demo.py

[qiqiApink]

I run the demo.py with python demo.py --ckpt data/ckpt/hr48-PA43.0_MJE69.0_MVE81.2_3dpw.pt --backbone hr48 --input_path test_samples/downtown_runForBus_00_3dpw.mp4 --input_type video --save_results --make_video --frame_rate 30

But I got the "Segmentation fault" like this:

--------------------------- Visualization ---------------------------
Front view directory: /mnt/lustre/zhangyaqi/noah-research/CLIFF/test_samples/downtown_runForBus_00_3dpw/front_view_hr48
  0%|          | 0/731 [00:00<?, ?it/s]srun: error: SH-IDC1-10-140-1-165: task 0: Segmentation fault (core dumped)

How can I solve this?

[zhihaolee]

Sorry I don't know, maybe it's the library problem in your enviroment.

[AllendLi]

Modifying the torch version with the following command may be useful
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113