【深度学习】web版一键训练系统

动机

由于工作需要,常常会需要对模型进行日常训练更新,基本的pipeline包括:

  • dump新的数据
  • 处理数据
  • 训练网络模型
  • 处理模型
  • 测试模型
  • 发布

这些工作其实主要是按部就班的事情,处理数据和网络结构等不会每天都改动。所以一直先做也个傻瓜版本的训练系统,直接交给工程人员使用,这样一个人就可以一键完成整个pipeline,从而大大减少人力成为,并减少由于误操作带来的问题。

特性

这个网页版训练系统应该包括如下特性那个:

  • 数据管理功能(上传、选择训练集和测试集)
  • 一键训练和测试(实时显示进度,对测试结果进行可视化)
  • 一键发布(可选功能)

搁浅

但是由于一直太忙,加上训练代码也经常变动作为理由,一直没有着手做这件事情。

启发

直到今天看paper时,无意间发现了一个NB的工作——CloudCV,其中便包含了和我要的类似的功能,而且论文对系统的架构和使用的工具进行了较为细致的说明。再次激发了我做一个web版一键训练系统的冲动,准备逐步学习论文中提到的工具。

你可能感兴趣的:(深度学习)