tainc commited on
Commit
a8a0e75
1 Parent(s): 5a8a37a

update links from sgnlp to sgnlp-models

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