Function _make_conv

test/network/test_mstdp_florian.py:381–401

(conn_cls, in_shape, out_shape, rule, batch, tc=20.0)

def _make_conv(conn_cls, in_shape, out_shape, rule, batch, tc=20.0):
    src = Input(shape=in_shape)
    tgt = LIFNodes(shape=out_shape)
    src.batch_size = batch
    tgt.batch_size = batch
    src.s = torch.zeros(batch, *in_shape, dtype=torch.bool)
    tgt.s = torch.zeros(batch, *out_shape, dtype=torch.bool)
    conn = conn_cls(
        src,
        tgt,
        kernel_size=3,
        stride=1,
        nu=1e-2,
        update_rule=rule,
        tc_plus=tc,
        tc_minus=tc,
    )
    conn.dt = 1.0
    with torch.no_grad():
        conn.w.zero_()
    return conn


class TestConvMSTDP: