espnet.nets.pytorch_backend.transformer.plot.PlotAttentionReport

source

class espnet.nets.pytorch_backend.transformer.plot.PlotAttentionReport(att_vis_fn, data, outdir, converter, transform, device, reverse=False, ikey='input', iaxis=0, okey='output', oaxis=0, subsampling_factor=1)

Bases: PlotAttentionReport

Plot Attention Report class.

Initialize Plot Attention Report.

get_attention_weights()

Get attention weights.

log_attentions(logger, step)

Log attentions.

plotfn(*args, **kwargs)

Process plot function.