DependableSystemsLab/TensorFI

Vgg11 dataset not working on python2

zoythum opened this issue · 1 comments

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

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)