Wrapper for PyTorch models - Githubissues

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

Wrapper for PyTorch models #98

Open abheesht17 opened 2 years ago

abheesht17 commented 2 years ago

Similar to https://github.com/Adversarial-Deep-Learning/code-soup/issues/97.

Reference: https://github.com/QData/TextAttack/blob/master/textattack/models/wrappers/pytorch_model_wrapper.py