mxnet
DevinDujun
这个作者很懒,什么都没留下…
展开
-
mxnet 多卡训练
mxnet多卡训练 gluon单卡训练多卡训练补充参考blog 单卡训练 devices = mx.gpu(0) #指定GPU #batch_data,batch_label 表示从生成器中获取的数据和label #将nd.array类型数据放到devices(指定GPU(0,1,2,3)或者cpu)上 data = mx.nd.array(batch_data).as_in_context(devices) label = mx.nd.array(batch_label).as_in_context(原创 2020-12-09 22:59:16 · 473 阅读 · 0 评论 -
mxnet symbol 和gluon 模型加载
mxnet symbol 和gluon 模型加载常规流程数据准备定义精度评价函数定义网络训练symbol与Gluon 模型加载gluon保存和加载模型结构和参数参考博客 常规流程 使用mxnet搭建模型模型的流程:主要包括数据准备,模型构建,超级参数选择,优化器,loss评价标准 数据准备 from mxnet import gluon from mxnet.gluon import nn import matplotlib.pyplot as plt from mxnet import autograd原创 2020-12-09 00:06:43 · 539 阅读 · 0 评论