先谈谈操作体验:整个过程只需点点点即可
时间:2022-03-26 17:57:01 | 来源:行业动态
时间:2022-03-26 17:57:01 来源:行业动态
培训课程采用的是云端 NVIDIA GPU,具体基于NVIDIA的深度学习训练系统(Deep Learning GPU Training System, DIGITS)完成。由于系统中封装了目前所流行的大部分深度学习开发框架,包括Caffe2、Torch、TensorFlow、MXNet等。学员只需要点点鼠标、改改参数,就能轻松实现深度学习训练。
以图像分类为例,整个动手实验记者只做了以下操作:数据集创建、模型创建、模型训练、模型测试,就完成了基于深度学习模型的图像分类、检测等任务。
实验中所用的是MINIST数据集,由0-9共10个数字的不同手写文字的图像组成,每个手写数字都是28*28像素的图像,共有70000个训练样本和10000个测试样本。
通过将数据输入到创建的模型中(这里使用的是DIGITS现有的LeNet卷积神经网络),在1分钟内就完成了对70000个样本的训练。同时,整个过程以可视化界面的方式展现出来,可以随时看到训练损失、验证损失、验证正确率、学习率变化、GPU资源占用等训练进度信息。如下图:
模型训练完成后,就可以导入手写体数字图片进行测试,以下是测试结果,呈现了图片为具体数字可能性的概率:
完成!总而言之,整个过程只需点点点即可。
当然,这只是NVIDIA DLI系列培训中非常基础的一个课程,开发者在实际应用中也可以根据需要导入自己的数据集、创建自己的模型,去完成如图像分割、图像生成,以及自然语言处理等更深入的场景应用。