tainc commited on
Commit
c952bef
·
1 Parent(s): aef9cae

updated links from sgnlp to sgnlp-models

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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