flair.trainers.plugins.ReduceTransformerVocabPlugin#

class flair.trainers.plugins.ReduceTransformerVocabPlugin(base_path, save_optimizer_state)View on GitHub#

Bases: TrainerPlugin

__init__(base_path, save_optimizer_state)View on GitHub#

Initialize the base plugin.

Methods

__init__(base_path, save_optimizer_state)

Initialize the base plugin.

attach_to(pluggable)

Attach this plugin to a Pluggable.

detach()

Detach a plugin from the Pluggable it is attached to.

get_state()

hook([first_arg])

Convience function for BasePlugin.mark_func_as_hook).

mark_func_as_hook(func, *events)

Mark method as a hook triggered by the Pluggable.

register_transformer_smaller_training_vocab(**kw)

save_model_at_the_end(**kw)

Attributes

attach_to_all_processes

If set, the plugin will be attached to all processes when distributed, not just the main process.

corpus

model

pluggable

trainer

register_transformer_smaller_training_vocab(**kw)View on GitHub#
save_model_at_the_end(**kw)View on GitHub#
property attach_to_all_processes: bool#

If set, the plugin will be attached to all processes when distributed, not just the main process.