ask for help

[zcdliuwei]

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

[Boese0601]

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

[zcdliuwei]

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

[Boese0601]

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.