paddlespeech.t2s.training.extensions.visualizer module

class paddlespeech.t2s.training.extensions.visualizer.VisualDL(logdir)

Bases: Extension

A wrapper of visualdl log writer. It assumes that the metrics to be visualized are all scalars which are recorded into the .observation dictionary of the trainer object. The dictionary is created for each step, thus the visualdl log writer uses the iteration from the updater's iteration as the global step to add records.

Attributes:

name

Methods

__call__(trainer)	Main action of the extention.
finalize(trainer)	Action that is executed when training is done.
initialize(trainer)	Action that is executed once to get the corect trainer state.
on_error(trainer, exc, tb)	Handles the error raised during training before finalization.

default_name = 'visualdl'

finalize(trainer)

Action that is executed when training is done. For example, visualizers would need to be closed.

priority = 100

trigger = (1, 'iteration')