---
_import: ../lenet5_bn.yaml
_trainable: true
_gate:
    enable: true
    policy: parametric
    density: 1.0
    pool: avg
    weight: 0.000001
    regularizer:
        l1: 0.00000001
        moe: 0.0001
    threshold: global
model:
    name: lenet5
    description: |
        A variant of cifarnet that performs online gating.
    layers:
        _gated_conv: &gated_conv
            type: gated_convolution
            gate_params: $(_gate)
            trainable: $(_trainable)
        conv0: {<<: *gated_conv}
        conv1: {<<: *gated_conv}
