spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10 package

Submodules

spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet module

ResNet in PyTorch.

For Pre-activation ResNet, see ‘preact_resnet.py’.

Reference: [1] Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun

Deep Residual Learning for Image Recognition. arXiv:1512.03385

class spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet.BasicBlock(in_planes, planes, stride=1)[源代码]

基类:torch.nn.modules.module.Module

expansion = 1
forward(x)[源代码]
training: bool
class spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet.Bottleneck(in_planes, planes, stride=1)[源代码]

基类:torch.nn.modules.module.Module

expansion = 4
forward(x)[源代码]
training: bool
class spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet.ResNet(block, num_blocks, num_classes=10)[源代码]

基类:torch.nn.modules.module.Module

forward(x)[源代码]
training: bool
spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet.ResNet18()[源代码]
spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet.ResNet34()[源代码]
spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet.ResNet50()[源代码]
spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet.ResNet101()[源代码]
spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet.ResNet152()[源代码]
spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.resnet.test()[源代码]

spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.vgg module

VGG11/13/16/19 in Pytorch.

class spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.vgg.VGG(vgg_name)[源代码]

基类:torch.nn.modules.module.Module

forward(x)[源代码]
training: bool
class spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.vgg.VGG_no_bias_bn(vgg_name)[源代码]

基类:torch.nn.modules.module.Module

forward(x)[源代码]
training: bool
spikingjelly.clock_driven.ann2snn.examples.model_sample.cifar10.vgg.test()[源代码]

Module contents