ensemble

Classes

ensemble through model averaging.

Module Contents

class ensemble.Ensemble(modelA, modelB, modelC=None)

Bases: torch.nn.Module

ensemble through model averaging.

name

modelA

modelB

modelC

forward_head_batch(*, e1_idx, rel_idx)