Unbabel/OpenKiwi

View on GitHub
docs/configuration/train.rst

Summary

Maintainability
Test Coverage
Training configuration
======================

There are training configuration files available in ``config/`` for all supported models.

As an example, here is the configuration for the ``Bert`` model:

.. literalinclude:: ../../config/bert.yaml
   :language: yaml


Configuration class
-------------------

Full API reference: :class:`kiwi.lib.train.Configuration`


.. autosummary:
   :toctree: stubs

..   kiwi.lib.train.Configuration
   kiwi.lib.train.RunConfig
   kiwi.lib.train.TrainerConfig
   kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset.Config
   kiwi.systems.qe_system.QESystem.Config


.. autoclass:: kiwi.lib.train.Configuration
   :noindex: