首先,要进行ANN的训练,训练出来的一个文件模型是pkl格式的模型。然后试图将这个ANN模型转换为SNN
这个ANN的模型的网络结构:三层卷积块串联一个全连接层,每个卷积块由一个卷积层、一个批正则化、一个ReLU激活和一个平均池化层组成
代码如下:
self.network = nn.Sequential(
nn.Conv2d(1, 32, 3, 1),
nn.BatchNorm2d(32
首先,要进行ANN的训练,训练出来的一个文件模型是pkl格式的模型。然后试图将这个ANN模型转换为SNN
这个ANN的模型的网络结构:三层卷积块串联一个全连接层,每个卷积块由一个卷积层、一个批正则化、一个ReLU激活和一个平均池化层组成
代码如下:
self.network = nn.Sequential(
nn.Conv2d(1, 32, 3, 1),
nn.BatchNorm2d(32