Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

Commit

Permalink
Add paper link (#4605)
Browse files Browse the repository at this point in the history
  • Loading branch information
jaseweston authored Jun 16, 2022
1 parent d7674e9 commit d83cb03
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions projects/director/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

Kushal Arora, Kurt Shuster, Sainbayar Sukhbaatar, Jason Weston

<!-- Paper: https://arxiv.org/2206. -->
<b>Paper Link</b>: [https://arxiv.org/2206](https://arxiv.org/abs/2206.07694)


## Abstract
Expand All @@ -27,4 +27,4 @@ parlai train -vtim 300 -bs 6 --gradient-clip 10.0 --fp16 True -lr 1e-05 --valida
### Evaluate DIRECTOR model on the toxic prompts from WikiToxicComments dataset:
```python
python -m parlai.scripts.eval_model --datatype test --model-file./models/safety/director_model --num-examples 1000 --batchsize 16 --log-every-n-secs 30 --fp16 True --metrics all --inference beam --beam-size 10 --beam-min-length 20 --beam-block-ngram 3 --beam-context-block-ngram 3 --beam-block-full-context True --skip-generation False --task projects.director.tasks.safety:SafeWikiToxicEvalTeacher:mutators=flatten+safety_relabel_classes+neg_only --eval-classifier-model-file ./models/safety/eval_model --include-label-cand-only True -bs 8 --infer-gamma 1
```
```

0 comments on commit d83cb03

Please sign in to comment.