颜色分类leetcode pytorch experiments template 基于pytorch的分类实验模板
颜色分类leetcode Pytorch分类实验模板使用pytorch并进行分类的深度学习项目模板。该模板被设计为一个完全有效的实验启动器。也就是说,简单地运行python train.py将在Cifar-10上运行一个小型CNN,同时处理日志记录、检查点、整齐打印到终端、数据集等。一些值得注意的功能包括:立即使用以下型号:一个小/浅的CNN标准的Resnet预激活Resnet广泛的Resnet密集网以下数据集的立即可用性: -放大(即Cinic-10的train+validation作为train set) Cifar-100 MNIST时尚-MNIST内置日志记录和进度条内置和广泛的数据增强,包括:随机水平翻转随机作物剪下随机限制旋转随机缩放随机剪切颜色抖动A举例说明简单日志记录功能的使用文件夹/文件结构train.py :主要训练代码,运行这个→→→ models :神经网络→→→ notebooks :绘图结果的笔记本→→→ utils :必要的实用程序,包括实验管理员和数据集示例运行
用户评论