OpenMMLab Day03

MMClassification

图像分类工具包使用
OpenMMLab Day03_第1张图片

环境

OpenMMLab Day03_第2张图片

配置文件说明

深度学习模型的训练涉及几个方面:

  • 模型结构:模型有几层、每层多少通道数等等
  • 数据集:用什么数据训练模型:数据集划分、数据文件路径、数据增强策略等等
  • 训练策略:梯度下降算法、学习率参数、batch_size、训练总轮次、学习率变化策略等等
  • 运行时:GPU、分布式环境配置等等
  • 一些辅助功能:如打印日志、定时保存checkpoint等等

在 OpenMMLab 项目中,所有这些项目都涵盖在一个配置文件中,一个配置文件定义了一个完整的训练过程

  • model:定义模型
  • data:定义数据
  • optimizer、lr_config:定义训练策略
  • load_from:定义与训练模型的参数文件

配置文件运行机制

OpenMMLab Day03_第3张图片

你可能感兴趣的:(训练营,深度学习,人工智能,计算机视觉,分类)