Seeding/Checkpoints for Models / Attacks

Adversarial-Deep-Learning / code-soup

This is a collection of algorithms and approaches used in the book adversarial deep learning

MIT License

18 stars 18 forks source link

Seeding/Checkpoints for Models / Attacks #60

Closed someshsingh22 closed 2 years ago

someshsingh22 commented 2 years ago

[ ] Seeding all random variables, random, torch, numpy
[ ] Save and Load model checkpoints using torch.save

soham-chitnis10 commented 2 years ago

Hi I would like to take up this issue My approach : Function to save a model which takes inputs as model and path with checkpoint to save and function to load from a checkpoint and return the model. Function for seeding all the random variables.

someshsingh22 commented 2 years ago

@soham-chitnis10 You can take this up however instead of function use classes.