EnsembleLinear.Config(
        channels_in=4,
        channels_out=3,
        num_ensemble=2,
        bias=False,
        depth_index=(),
        init_weight=priml.model.init.kaiming_uniform,
        shard=None
    )
