Saving and serializing AttackModelBundle

spring-epfl / mia

A library for running membership inference attacks against ML models

MIT License

138 stars 28 forks source link

Saving and serializing AttackModelBundle #10

Open Zeltserj opened 4 years ago

Zeltserj commented 4 years ago

I ran the CIFAR10 example and I would like to save it so I can run some tests without having to retrain the attack model every time and also be able to use it elsewhere. I can't do it conveniently. Is there a way to do so?

I have tried pickle.dump() , as well as using _get_model() and the saving it.

Thanks