I am trying to re-train the model (solution 4) but when i save the model it saves as zip file. I have pre-trained model files from original author.
Is there any way to save the file to .pth but in Binary format?
torch.save
has an undocumented/not guaranteed option _use_new_zipfile_serialization
that might help.
However, I would venture that it is better to use the new format. What is not working for you with it?
Best regards
Thomas
_use_new_zipfile_serialization = False
worked for me. I found it somewhere else.
Thanks