PyTorch图片分类:CIFAR-10
目录
1.注意
基于PyTorch的CIFAR-10图像分类。
CNN模型是ResNet-18。
该存储库受PyTorch模板项目[1]和带有PyTorch的Train CIFAR10 [2]的启发。
但是,该存储库已与PyTorch模板项目分离,以便集中精力快速研究和开发高级功能,而无需考虑向后兼容性。 特别是,与PyTorch模板项目不同,您可以使用dataloader /轻松处理自己的数据集。
我建议您应忽略带有八字体#的注释说明。
修改日期:2021年3月31日。
2.总结开发环境
操作系统(OS):Ubuntu MATE 18.04.3 LTS(Bionic)
图形处理单元(GPU):NVIDIA TITAN Xp,1ea
GPU驱动程序:Nvidia-450.102.04
CUDA工具包:CUDA 10.2
cuDNN
2021-11-07 22:09:38
80.43MB
Python
1