文件名称:pytorch-cifar10:pytorch和cifar10
文件大小:14KB
文件格式:ZIP
更新时间:2024-05-21 03:41:36
pytorch lenet densenet resnet alexnet
pytorch-cifar10 使用PyTorch在CIFAR10上的个人实践灵感来自来自 。 介绍 CIFAR-10数据集包含10个类别的60000个32x32彩色图像,每个类别6000个图像。 有50000张训练图像和10000张测试图像。 数据集分为五个训练批次和一个测试批次,每个批次具有10000张图像。 测试批次包含每个类别中恰好1000个随机选择的图像。 训练批次按随机顺序包含其余图像,但是某些训练批次可能包含比另一类更多的图像。 在它们之间,培训批次包含每个班级的正好5000张图像。 要求 python3.6 麻木 火炬0.4.0 火炬视觉0.2.0 用法 python3 main.py 可选参数: --lr default=1e-3 learning rate --epoch default=200
【文件预览】:
pytorch-cifar10-master
----misc.py(2KB)
----models()
--------VGG.py(1KB)
--------AlexNet.py(1KB)
--------ResNet.py(4KB)
--------__init__.py(160B)
--------GoogleNet.py(3KB)
--------LeNet.py(680B)
--------WideResNet.py(4KB)
--------DenseNet.py(3KB)
----main.py(6KB)
----LICENSE(11KB)
----README.md(3KB)
----.gitignore(40B)