One-Thing-One-Click icon indicating copy to clipboard operation
One-Thing-One-Click copied to clipboard

Notice on data preparation

Open Kaisyouu opened this issue 2 years ago • 0 comments

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.

Kaisyouu avatar Apr 03 '22 05:04 Kaisyouu