kiwi.systems.xlmroberta

Module Contents

Classes

ModelConfig	Base class for all pydantic configs. Used to configure base behaviour of configs.
XLMRoberta	XLMRoberta-based Predictor-Estimator model

kiwi.systems.xlmroberta.logger

class kiwi.systems.xlmroberta.ModelConfig

Bases: kiwi.utils.io.BaseConfig

Base class for all pydantic configs. Used to configure base behaviour of configs.

encoder :XLMRobertaEncoder.Config

decoder :LinearDecoder.Config

outputs :QEOutputs.Config

tlm_outputs :TLMOutputs.Config

class kiwi.systems.xlmroberta.XLMRoberta(config, data_config: WMTQEDataset.Config = None, module_dict: Dict[str, Any] = None)

Bases: kiwi.systems.qe_system.QESystem

XLMRoberta-based Predictor-Estimator model

class Config

Bases: kiwi.systems.qe_system.QESystem.Config

System configuration base class.

model :ModelConfig

kiwi.systems.xlm kiwi.training