文件名称：PyTorch_CIFAR10:CIFAR10数据集上的预训练TorchVision模型（带有权重）
文件大小：22KB
文件格式：ZIP
更新时间：2024-06-11 01:27:41
computer-vision pytorch deeplearning pretrained-models cifar10
在CIFAR-10数据集上训练的PyTorch模型 我修改了官方实施的流行CNN模型，并对CIFAR-10数据集进行了训练。 我在原始代码中更改了类的数量，过滤器大小，步幅和填充，以便它可以与CIFAR-10一起使用。 我也共享这些模型的权重，因此您只需加载权重并使用它们即可。 通过使用PyTorch-Lightning，该代码具有很高的可复制性和可读性。 支持模型的统计 不。 模型 值累积 否。参数 尺寸 1个 vgg11_bn 92.39％ 28.150百万 108兆字节 2个 vgg13_bn 94.22％ 28.334百万 109兆字节 3 vgg16_bn 94.00％ 33.647百万 129兆字节 4 vgg19_bn 93.95％ 38.959百万 149兆字节 5 网路18 93.07％ 11.174百万 43兆字节 6 资源34 9
【文件预览】：
PyTorch_CIFAR10-master
----train.py(3KB)
----data.py(3KB)
----LICENSE(1KB)
----README.md(3KB)
----module.py(2KB)
----.gitignore(2KB)
----cifar10_models()
--------resnet_orig.py(3KB)
--------mobilenetv2.py(4KB)
--------vgg.py(5KB)
--------googlenet.py(8KB)
--------inception.py(13KB)
--------densenet.py(8KB)
--------resnet.py(9KB)
----schduler.py(5KB)