颜色分类leetcode pytorch experiments template:基于pytorch的分类实验模板

颜色分类leetcode Pytorch分类实验模板使用pytorch并进行分类的深度学习项目模板。该模板被设计为一个完全有效的实验启动器。也就是说，简单地运行python train.py将在Cifar-10上运行一个小型CNN，同时处理日志记录、检查点、整齐打印到终端、数据集等。一些值得注意的功能包括：

立即使用以下型号：

• 一个小/浅的CNN

• 标准的Resnet

• 预激活Resnet

• 广泛的Resnet

• 密集网

以下数据集的立即可用性：

• Cifar-100

• MNIST

• 时尚-MNIST

内置日志记录和进度条

内置和广泛的数据增强，包括：

• 随机水平翻转

• 随机作物

• 剪下

• 随机限制旋转

• 随机缩放

• 随机剪切

• 颜色抖动

举例说明简单日志记录功能的使用：

文件夹/文件结构：

• train.py :主要训练代码，运行这个

• models ：神经网络

• notebooks :绘图结果的笔记本

• utils ：必要的实用程序，包括实验管理员和数据集示例运行