update links from sgnlp to sgnlp-models
Browse files
README.md
CHANGED
@@ -180,16 +180,16 @@ preprocessor = RstPreprocessor()
|
|
180 |
postprocessor = RstPostprocessor()
|
181 |
|
182 |
segmenter_config = RstPointerSegmenterConfig.from_pretrained(
|
183 |
-
'https://storage.googleapis.com/sgnlp/models/rst_pointer/segmenter/config.json')
|
184 |
segmenter = RstPointerSegmenterModel.from_pretrained(
|
185 |
-
'https://storage.googleapis.com/sgnlp/models/rst_pointer/segmenter/pytorch_model.bin',
|
186 |
config=segmenter_config)
|
187 |
segmenter.eval()
|
188 |
|
189 |
parser_config = RstPointerParserConfig.from_pretrained(
|
190 |
-
'https://storage.googleapis.com/sgnlp/models/rst_pointer/parser/config.json')
|
191 |
parser = RstPointerParserModel.from_pretrained(
|
192 |
-
'https://storage.googleapis.com/sgnlp/models/rst_pointer/parser/pytorch_model.bin',
|
193 |
config=parser_config)
|
194 |
parser.eval()
|
195 |
|
@@ -217,15 +217,15 @@ trees = postprocessor(sentences=sentences, tokenized_sentences=tokenized_sentenc
|
|
217 |
|
218 |
# Training
|
219 |
The dataset (RST Discourse Treebank) that the model is trained on is a licensed dataset.
|
220 |
-
- **Training Config:** [Segmenter](https://storage.googleapis.com/sgnlp/models/rst_pointer/segmenter/training_config.json) | [Parser](https://storage.googleapis.com/sgnlp/models/rst_pointer/parser/training_config.json)
|
221 |
|
222 |
#### Training Results
|
223 |
- **Training Time (Segmenter):** ~2 hours for 100 epochs on a single V100 GPU for segmenter model.
|
224 |
- **Training Time (Parser):** ~6 hours for 200 epochs on a single V100 GPU for parser model
|
225 |
|
226 |
# Model Parameters
|
227 |
-
- **Model Weights:** [Segmenter](https://storage.googleapis.com/sgnlp/models/rst_pointer/segmenter/pytorch_model.bin) | [Parser](https://storage.googleapis.com/sgnlp/models/rst_pointer/parser/pytorch_model.bin)
|
228 |
-
- **Model Config:** [Segmenter](https://storage.googleapis.com/sgnlp/models/rst_pointer/segmenter/config.json) | [Parser](https://storage.googleapis.com/sgnlp/models/rst_pointer/parser/config.json)
|
229 |
- **Model Inputs:** A sentence.
|
230 |
- **Model Outputs:** Discourse parsed tree.
|
231 |
- **Model Size:** ~362MB for segmenter model, ~361MB for parser model
|
|
|
180 |
postprocessor = RstPostprocessor()
|
181 |
|
182 |
segmenter_config = RstPointerSegmenterConfig.from_pretrained(
|
183 |
+
'https://storage.googleapis.com/sgnlp-models/models/rst_pointer/segmenter/config.json')
|
184 |
segmenter = RstPointerSegmenterModel.from_pretrained(
|
185 |
+
'https://storage.googleapis.com/sgnlp-models/models/rst_pointer/segmenter/pytorch_model.bin',
|
186 |
config=segmenter_config)
|
187 |
segmenter.eval()
|
188 |
|
189 |
parser_config = RstPointerParserConfig.from_pretrained(
|
190 |
+
'https://storage.googleapis.com/sgnlp-models/models/rst_pointer/parser/config.json')
|
191 |
parser = RstPointerParserModel.from_pretrained(
|
192 |
+
'https://storage.googleapis.com/sgnlp-models/models/rst_pointer/parser/pytorch_model.bin',
|
193 |
config=parser_config)
|
194 |
parser.eval()
|
195 |
|
|
|
217 |
|
218 |
# Training
|
219 |
The dataset (RST Discourse Treebank) that the model is trained on is a licensed dataset.
|
220 |
+
- **Training Config:** [Segmenter](https://storage.googleapis.com/sgnlp-models/models/rst_pointer/segmenter/training_config.json) | [Parser](https://storage.googleapis.com/sgnlp-models/models/rst_pointer/parser/training_config.json)
|
221 |
|
222 |
#### Training Results
|
223 |
- **Training Time (Segmenter):** ~2 hours for 100 epochs on a single V100 GPU for segmenter model.
|
224 |
- **Training Time (Parser):** ~6 hours for 200 epochs on a single V100 GPU for parser model
|
225 |
|
226 |
# Model Parameters
|
227 |
+
- **Model Weights:** [Segmenter](https://storage.googleapis.com/sgnlp-models/models/rst_pointer/segmenter/pytorch_model.bin) | [Parser](https://storage.googleapis.com/sgnlp-models/models/rst_pointer/parser/pytorch_model.bin)
|
228 |
+
- **Model Config:** [Segmenter](https://storage.googleapis.com/sgnlp-models/models/rst_pointer/segmenter/config.json) | [Parser](https://storage.googleapis.com/sgnlp-models/models/rst_pointer/parser/config.json)
|
229 |
- **Model Inputs:** A sentence.
|
230 |
- **Model Outputs:** Discourse parsed tree.
|
231 |
- **Model Size:** ~362MB for segmenter model, ~361MB for parser model
|