flair.trainers.plugins.WeightExtractorPlugin#

class flair.trainers.plugins.WeightExtractorPlugin(base_path)View on GitHub #

Simple Plugin for weight extraction.

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.
`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.

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`

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.