storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Introduce code from Rust monorepo. This is a mess

Browse Source
This commit is contained in:
Brandon Thomas
2022-02-28 05:43:54 -05:00
parent 921fb49f9a
commit ba326bfcfd
1058 changed files with 192192 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/NeMo/docs/source/nlp/api.rst
Executable
+74
View File
@@ -0,0 +1,74 @@
NeMo NLP collection API
=======================
Model Classes
-------------
.. autoclass:: nemo.collections.nlp.models.TextClassificationModel
:show-inheritance:
:members: setup_training_data, setup_optimization, setup_validation_data, setup_test_data, register_artifact, classifytext
.. autoclass:: nemo.collections.nlp.models.GLUEModel
:show-inheritance:
:members: setup_training_data, setup_optimization, setup_validation_data, setup_test_data, register_artifact
.. autoclass:: nemo.collections.nlp.models.PunctuationCapitalizationModel
:show-inheritance:
:members: add_punctuation_capitalization, setup_training_data, setup_optimization, setup_validation_data, setup_test_data, multi_validation_epoch_end, register_artifact
.. autoclass:: nemo.collections.nlp.models.TokenClassificationModel
:show-inheritance:
:members: setup_training_data, setup_optimization, setup_validation_data, setup_test_data, register_artifact
.. autoclass:: nemo.collections.nlp.models.QAModel
:show-inheritance:
:members: setup_training_data, setup_optimization, setup_validation_data, setup_test_data, inference, validation_epoch_end, test_epoch_end
.. autoclass:: nemo.collections.nlp.models.BERTLMModel
:show-inheritance:
:members: setup_training_data, setup_optimization
Modules
-------
.. autoclass:: nemo.collections.nlp.modules.BertModule
:show-inheritance:
:members:
.. autoclass:: nemo.collections.nlp.modules.MegatronBertEncoder
:show-inheritance:
:members:
.. autoclass:: nemo.collections.nlp.modules.AlbertEncoder
:show-inheritance:
:members:
.. autoclass:: nemo.collections.nlp.modules.BertEncoder
:show-inheritance:
:members:
.. autoclass:: nemo.collections.nlp.modules.DistilBertEncoder
:show-inheritance:
:members:
.. autoclass:: nemo.collections.nlp.modules.RobertaEncoder
:show-inheritance:
:members:
.. autoclass:: nemo.collections.nlp.modules.SequenceClassifier
:show-inheritance:
:members:
.. autoclass:: nemo.collections.nlp.modules.SequenceRegression
:show-inheritance:
:members:
.. autoclass:: nemo.collections.nlp.modules.SequenceTokenClassifier
:show-inheritance:
:members:
.. autofunction:: nemo.collections.nlp.modules.get_lm_model
.. autofunction:: nemo.collections.nlp.modules.get_pretrained_lm_models_list
.. autofunction:: nemo.collections.nlp.modules.get_megatron_lm_models_list