SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[pplm] Update S3 links

Browse Source 
Co-Authored-By: Piero Molino <w4nderlust@gmail.com>
This commit is contained in:
Julien Chaumond
2019-12-03 10:53:02 -05:00
parent 96e83506d1
commit f434bfc623
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/pplm/run_pplm.py
View File

@@ -59,7 +59,7 @@ BAG_OF_WORDS_ARCHIVE_MAP = {
DISCRIMINATOR_MODELS_PARAMS = {
"clickbait": {
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/clickbait_classifierhead.pt",
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/clickbait_classifier_head.pt",
"class_size": 2,
"embed_size": 1024,
"class_vocab": {"non_clickbait": 0, "clickbait": 1},
@@ -67,7 +67,7 @@ DISCRIMINATOR_MODELS_PARAMS = {
"pretrained_model": "gpt2-medium",
},
"sentiment": {
"url": "http://s.yosinski.com/SST_classifier_head.pt",
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/SST_classifier_head.pt",
"class_size": 5,
"embed_size": 1024,
"class_vocab": {"very_positive": 2, "very_negative": 3},
@@ -75,7 +75,7 @@ DISCRIMINATOR_MODELS_PARAMS = {
"pretrained_model": "gpt2-medium",
},
"toxicity": {
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/toxicity_classifierhead.pt",
"url": "https://s3.amazonaws.com/models.huggingface.co/bert/pplm/discriminators/toxic_classifier_head.pt",
"class_size": 2,
"embed_size": 1024,
"class_vocab": {"non_toxic": 0, "toxic": 1},