flair.trainers.plugins.WeightExtractorPlugin#
- class flair.trainers.plugins.WeightExtractorPlugin(base_path)View on GitHub#
Bases:
TrainerPlugin
Simple Plugin for weight extraction.
- __init__(base_path)View on GitHub#
Initialize the base plugin.
Methods
__init__
(base_path)Initialize the base plugin.
after_training_batch
(batch_no, epoch, ...)Extracts weights.
attach_to
(pluggable)Attach this plugin to a Pluggable.
detach
()Detach a plugin from the Pluggable it is attached to.
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.
Attributes
If set, the plugin will be attached to all processes when distributed, not just the main process.
corpus
model
pluggable
trainer
- after_training_batch(batch_no, epoch, total_number_of_batches, **kw)View on GitHub#
Extracts weights.
- property attach_to_all_processes: bool#
If set, the plugin will be attached to all processes when distributed, not just the main process.
- get_state()View on GitHub#
- Return type:
dict
[str
,Any
]