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
|