Closed harlanhong closed 1 year ago
That is weird since there is no error reported. Maybe it is processing the 3DMM extraction. You can print log in inner modules to check which code line it stuck at. By the way, we recommend using A100 or 3090 to run the code.
did you solve it?
Hi,
Thanks for your nice work. After I install StyleHEAT and run the command:
CUDA_VISIBLE_DEVICES=0 python inference.py --config configs/inference.yaml --video_source=./docs/demo/videos/RD_Radio34_003_512.mp4 -output_dir=./docs/demo/output --if_extract
The process freezes as shown in the following picture:
How to solve this problem?