updated links from sgnlp to sgnlp-models
Browse files
README.md
CHANGED
@@ -106,11 +106,11 @@ from sgnlp.models.emotion_entailment import (
|
|
106 |
|
107 |
# Load model
|
108 |
config = RecconEmotionEntailmentConfig.from_pretrained(
|
109 |
-
"https://storage.googleapis.com/sgnlp/models/reccon_emotion_entailment/config.json"
|
110 |
)
|
111 |
tokenizer = RecconEmotionEntailmentTokenizer.from_pretrained("roberta-base")
|
112 |
model = RecconEmotionEntailmentModel.from_pretrained(
|
113 |
-
"https://storage.googleapis.com/sgnlp/models/reccon_emotion_entailment/pytorch_model.bin",
|
114 |
config=config,
|
115 |
)
|
116 |
preprocessor = RecconEmotionEntailmentPreprocessor(tokenizer)
|
@@ -153,8 +153,8 @@ The train and evaluation datasets were derived from the RECCON dataset. The full
|
|
153 |
- **Training Time:** ~3 hours for 12 epochs on a single V100 GPU.
|
154 |
|
155 |
# Model Parameters
|
156 |
-
- **Model Weights:** [link](https://storage.googleapis.com/sgnlp/models/reccon_emotion_entailment/pytorch_model.bin)
|
157 |
-
- **Model Config:** [link](https://storage.googleapis.com/sgnlp/models/reccon_emotion_entailment/config.json)
|
158 |
- **Model Inputs:** Target utterance, emotion in target utterance, evidence utterance and conversational history.
|
159 |
- **Model Outputs:** Probability score of whether evidence utterance caused target utterance to exhibit the emotion specified.
|
160 |
- **Model Size:** ~477MB
|
|
|
106 |
|
107 |
# Load model
|
108 |
config = RecconEmotionEntailmentConfig.from_pretrained(
|
109 |
+
"https://storage.googleapis.com/sgnlp-models/models/reccon_emotion_entailment/config.json"
|
110 |
)
|
111 |
tokenizer = RecconEmotionEntailmentTokenizer.from_pretrained("roberta-base")
|
112 |
model = RecconEmotionEntailmentModel.from_pretrained(
|
113 |
+
"https://storage.googleapis.com/sgnlp-models/models/reccon_emotion_entailment/pytorch_model.bin",
|
114 |
config=config,
|
115 |
)
|
116 |
preprocessor = RecconEmotionEntailmentPreprocessor(tokenizer)
|
|
|
153 |
- **Training Time:** ~3 hours for 12 epochs on a single V100 GPU.
|
154 |
|
155 |
# Model Parameters
|
156 |
+
- **Model Weights:** [link](https://storage.googleapis.com/sgnlp-models/models/reccon_emotion_entailment/pytorch_model.bin)
|
157 |
+
- **Model Config:** [link](https://storage.googleapis.com/sgnlp-models/models/reccon_emotion_entailment/config.json)
|
158 |
- **Model Inputs:** Target utterance, emotion in target utterance, evidence utterance and conversational history.
|
159 |
- **Model Outputs:** Probability score of whether evidence utterance caused target utterance to exhibit the emotion specified.
|
160 |
- **Model Size:** ~477MB
|