espnet2.aqa.ar_universa.data.ARMetricCollateFn

source

class espnet2.aqa.ar_universa.data.ARMetricCollateFn(metric_token_pad_value=0, randomize=False)

Bases: CommonCollateFn

Pad metric/value pairs and optionally shuffle whole pairs for training.

Set the token padding ID and pair-order policy.