摘要:
读取和存储 在实际中,我们有时需要把训练好的模型部署到很多不同的设备。在这种情况下,我们可以把内存中训练好的模型参数存储在硬盘上供后续读取使用。 读写NDArray 可以直接使用save函数和load函数分别存储和读取NDArray 创建了NDArray变量x,并将其存在文件名同为x的文件里: #导 阅读全文
摘要:
自定义层 本节将介绍如何使用NDArray来自定义一个Gluon的层,从而可以被重复调用。 不含模型参数的自定义层 下面的CenteredLayer类通过继承Block类自定义了一个将输入减掉均值后输出的层,并将层的计算定义在了forward函数里。这个层里不含模型参数。 #导包 from mxne 阅读全文