由于工作需要,搭建了一套可视化的AI训练平台,初期先预置了常用的DNN,CNN,RNN算法,让负责业务模型的小伙伴们可以直接使用,不用自己去写代码了。支持多个任务并行运算。
模型结构的定制化、以及可视化部分参数的调参工作都在界面上配置完成,模型代码和神经网络层结构让后端自动生成,一键启动任务运行,查询返回结果!
训练完毕,再将model、Loss、accuracy打印输出到Web前端:
测试例比较简单,跑了一个50*50像素的图像分类任务,只启用了数据增强模式horizontal_flip,分别设置Dropout =0.5和0.7,过拟合的对比情况。
Dropout =0.5
Dropout =0.7
平台初步成型,刚刚起步,任重而道远!
领取专属 10元无门槛券
私享最新 技术干货