Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Notice on data preparation #19

Open
Kaisyouu opened this issue Apr 3, 2022 · 0 comments
Open

Notice on data preparation #19

Kaisyouu opened this issue Apr 3, 2022 · 0 comments

Comments

@Kaisyouu
Copy link

Kaisyouu commented Apr 3, 2022

Hi, all. @liuzhengzhe Kindly notice that there is something different in torch.save() between torch1.3 and torch1.6+.

Before doing python prepare_data_otoc.py, should modify line 128, add param_use_new_zipfile_serialization=False
Otherwise, torch.load('xxx.pth') will raise an error pytorch_model.bin is a zip archive (did you mean to use torch.jit.load()?)

torch1.3 has no param _use_new_zipfile_serialization=False so while preparing the data, your torch should remain ver1.6 or higher, and after generating .pth files, you should go back to torch1.3 or you will find troubles because PG_OP depends on torch1.3.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant