Vgg11 dataset not working on python2
zoythum opened this issue · 1 comments
zoythum commented
I am trying to use TensorFI with the provided vgg11 model, unfortunately when I try to train it I get the following error
ValueError: unsupported pickle protocol: 4
.
If I understood correctly it is a problem related to the pickle version used, could you provide me with the correct dataset?
Thank you
zoythum commented
For anyone having the same problem the solution is to convert the data using this snippet
#!/usr/bin/env python3
import pickle
with open("filename.p", "rb") as f:
w = pickle.load(f)
pickle.dump(w, open("filename_p2.p","wb"), protocol=2)