_yuki_

Transformer 系列：Vit Transformer、Swin Transformer

环境： pytorch 1.7.0 pytorch lightning 1.3.8MobileNetV1网络结构：代码复现：#2017import torchimport pytorch_lightning as plimport torch.nn as nndef ConvBN(in_channels,out_channels,kernel_size,stride=1,padding=0,bias=False,groups=1)...

SqueezeNet、Xception、ShuffleNetV1、ShuffleNetV2

环境： pytorch 1.7.0 pytorch lightning 1.3.8EfficientNetEfficientNet_V2

DenseNet、ResNeXt

Inception v4、Inception ResNet v1、Inception ResNet v2

Inception v2、Inception v3

环境： pytorch 1.7.0 pytorch lightning 1.3.8LeNet-5网络结构：代码复现：#1998import torchimport pytorch_lightning as plimport torch.nn as nnclass LeNet5(pl.LightningModule): def __init__(self): super().__init__() ...