上传者: 42175516
|
上传时间: 2021-11-08 11:37:54
|
文件大小: 14KB
|
文件类型: -
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