ask for help

[Keep Growing And Moving Forward]

First of all, thank you for selflessly sharing the code and pre-trained models
When I execute the following command： bash scripts/inference_tiktok_dataset.sh
I received an error message：

Is there a problem with my environment, such as the version of pytorch？
Looking forward to your reply

[Di Chang]

This is a path error, please double check your directory to the pretrained models. It should be “./pretrained_weights” instead of “pretrained_weighs”

[Keep Growing And Moving Forward]

Please ask again, as shown in the picture below, my parameter address seems to be correct：

[Di Chang]

I have updated inference_tiktok_dataset.sh. There's a typo in the image_pretrain_dir argument. The correct parameter should be "./pretrained_weights/model_state-110000.th". Thanks for your feedback.