Adversarial-Deep-Learning / Adversarial-Deep-Learning/code-soup

Wrapper for Hugging Face models

Aperta
#97 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub
good first issue Priority:Medium
Lingua principale
Jupyter Notebook
Stelle
17
Fork
17
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

Make a wrapper class for Hugging Face models. The class will initialise the model (and the tokeniser). Perform a forward pass on an input list of text. We need functions like `get_grad()` too, since some attacks require grads.

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

Guida per i contributori

Apri la guida per i contributori

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.