Skip to content

Instantly share code, notes, and snippets.

import sinabs
import sinabs.layers as sl
import torch
import torch.nn as nn
ann = nn.Sequential(
nn.Conv2d(1, 16, 5, bias=False),
nn.ReLU(),
nn.AvgPool2d(2),
nn.Conv2d(16, 32, 5, bias=False),
@biphasic
biphasic / connection_mask_bug.py
Last active September 30, 2020 09:51
sparse connection mask error
import numpy as np
import nxsdk.api.n2a as nx
net = nx.NxNet()
cxSrc = net.createCompartmentGroup(size=2, prototype=nx.CompartmentPrototype(biasMant=1000, biasExp=6))
cxDst = net.createCompartmentGroup(size=2, prototype=nx.CompartmentPrototype(vThMant=100))
prototypes = [nx.ConnectionPrototype(weightExponent=1),
nx.ConnectionPrototype(weightExponent=2)]
conn = cxSrc.connect(cxDst,