Skip to content

mean_fusion

Bases: metric_fusion

Source code in tinybig/fusion/metric_fusion.py 
72
73
74
class mean_fusion(metric_fusion):
    def __init__(self, name: str = "mean_fusion", *args, **kwargs):
        super().__init__(name=name, metric=batch_mean, *args, **kwargs)