ensemble

Module Contents

Classes

Ensemble

ensemble through model averaging.

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

Bases: torch.nn.Module

ensemble through model averaging.

forward_head_batch(*, e1_idx, rel_idx)