Saving and serializing AttackModelBundle

[Zeltserj]

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