Function _make_local

test/network/test_mstdp_florian.py:270–290

(lc, in_shape, out_shape, rule, batch)

def _make_local(lc, in_shape, out_shape, rule, batch):
    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 = lc(
        src,
        tgt,
        kernel_size=3,
        stride=1,
        n_filters=4,
        nu=1e-2,
        update_rule=rule,
        reduction=torch.sum,
    )
    conn.dt = 1.0
    with torch.no_grad():
        conn.w.zero_()
    return conn


class TestLocalMSTDPET: