Question about loading the pretrained salsanext model

[maneekwant]

Hi!

Thanks so much for the codes! I've a question about loading the pretrained salsanext model. When I followed the steps outlined in the "How to use" section and tested on the toy example, I came across the issue when trying to run infer.py on the toy dataset (python3 infer.py -d ../../../../data -m ../../../../data/model_salsanext_residual_1 -l ../../../../data/predictions_salsanext_residual_1_new -s valid) and got this error:

RuntimeError: ../../../../data/model_salsanext_residual_1/SalsaNext_valid_best is a zip archive (did you mean to use torch.jit.load()?)

I tried to switch from torch.load() to torch.jit.load() in the user.py as it suggested but it leads to other errors. What did I do wrong, or did I miss something along the way? I set up the environment according to the instruction linked on the github (using Pytorch 1.1).

Thank you in advance for your help!

[Chen-Xieyuanli]

Hey @maneekwant, thanks for using our code!

You need to unzip the pre-trained model before using it.

If there are still any problems, please feel free to let me know.

[maneekwant]

Thanks for the quick reply! I tried unzipping the SalsaNext_valid_best and now the error is that it’s a folder instead of a file. Did I miss something else too? Thanks so much!

[Chen-Xieyuanli]

Thanks for the quick reply! I tried unzipping the SalsaNext_valid_best and now the error is that it’s a folder instead of a file. Did I miss something else too? Thanks so much!

Sorry, I misunderstood your question. You need to only unzip model_salsanext_residual_1 but not SalsaNext_valid_best.

It seems your torch cannot load it and could be a version conflict. Mine is Version: 1.5.1+cu101. You may upgrade the pytorch and give it another try.

[maneekwant]

That works thanks so much!

[Chen-Xieyuanli]

Thank you for using our code! If you find the code and my answer helpful, I will appreciate it if you could star our repo :-)

[llxClover]

That works thanks so much!

@maneekwant ，I meet the same problem, "FileNotFoundError: [Errno 2] No such file or directory: '../../../../data/model_salsanext_residual_1/SalsaNext_valid_best'",,,,,,,
could you tell me how to solve?

[maneekwant]

@llxClover
I upgraded the pytorch version to 1.5.1 and that solved the issue!

[llxClover]

thanks！ I slao solved it by same way.

…

---Original--- From: ***@***.***> Date: Sat, Oct 23, 2021 04:12 AM To: ***@***.***>; Cc: ***@***.******@***.***>; Subject: Re: [PRBonn/LiDAR-MOS] Question about loading the pretrainedsalsanext model (#24) @llxClover I upgraded the pytorch version to 1.5.1 and that solved the issue! — You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android.