登录 注册
当前位置:主页 > 资源下载 > 9 > 基于pytorch的分类实验模板,颜色分类为leetcode-pytorch-experiments-template

基于pytorch的分类实验模板,颜色分类为leetcode-pytorch-experiments-template

  • 更新:2024-07-29 23:49:38
  • 大小:544KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:数据集 - 行业研究
  • 格式:ZIP

资源介绍

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