首页 / C# / 使用C#加载(预训练)的CNTK模型
使用C#加载(预训练)的CNTK模型
内容导读
互联网集市收集整理的这篇技术教程文章主要介绍了使用C#加载(预训练)的CNTK模型，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含3167字，纯文字阅读大概需要5分钟。
内容图文
我正在使用CNTK.GPU v2.2.0,并使用以下方法保存了模型：
model.Save(modelFilePath);
现在我想再次加载它,例如继续训练或只是评估样本.我可以看到两种方式.一种方法可行,但不可行.第二个不起作用.
>我再次从头开始构建神经网络的整个结构,然后在其上调用以下方法：
model.Restore(modelFilePath);
确实,这可行.
>我使用以下静态方法创建模型：
Function.Load(modelFilePath, DeviceDescriptor.GPUDevice(0));
这是行不通的.
完成这些操作后,我只为模型创建了一个训练器,创建了一个minibatchSource并尝试以与保存模型之前相同的方式训练模型.
但是使用第二种策略时,我得到以下异常：
System.ArgumentOutOfRangeException：’1个必需参数’Input(‘features’,[28 x 28 x 1],[,#])’的值,即请求的输出’Output(‘aggregateLoss’,[],[ ]),Output(‘lossFunction’,[1],[,#]),Output(‘aggregateEvalMetric’,[],[])’依赖,尚未提供.
[CALL STACK]
> CNTK::Internal:: UseSparseGradientAggregationInDataParallelSGD
- CNTK::Function:: Forward
- CNTK:: CreateTrainer
- CNTK::Trainer:: TotalNumberOfSamplesSeen
- CNTK::Trainer:: TrainMinibatch (x2)
- CSharp_CNTK_Trainer_TrainMinibatch__SWIG_0
- 00007FFA34AE8967 (SymFromAddr() error: The specified module could not be found.)
它说尚未提供输入功能.在训练和从头开始创建模型时,我正在使用输入：
var input = CNTKLib.InputVariable(_imageDimension, DataType.Float, _featureName);
var scaledInput = CNTKLib.ElementTimes(Constant.Scalar<float>(0.002953125f, _device), input);
...
因此,我认为我必须将加载的模型的输入替换为我为训练而创建的模型,并在从头开始创建模型时使用它-尽管输入没有不同.
但是我坚持尝试一下,因为我无法检索模型对象的输入,而我需要替换该输入(我认为).
model.FindByName(inputLayerName);
只是返回null,尽管我可以清楚地看到该名称与调试器中模型的“输入”列表中的图层名称匹配.
结果,我不知道如何正确加载保存的模型.我希望有一个人可以帮助我.
解决方法:
幸运的是,我自己找到了答案.我将其张贴在这里,因为可能还有其他CNTK初学者,他们可能会在这个问题上迷失方向,或者通常想知道如何正确加载模型.
问题是我没有使用相同的输入对象进行训练和模型创建.换句话说,如果让我通过提到的静态方法创建模型,我仍然必须确保模型中的对象与用于训练的对象相同.这应该通过以下方式实现：
>用您自己的输入对象替换已加载模型的输入,并将其用于训练.我没有测试,但是应该可以.
>提取已加载模型的输入并将其用于训练.我刚刚测试了它,它起作用了.
有我使用的代码：
var labels =
CNTKLib.InputVariable(new int[] {_classesNumber}, DataType.Float, _labelNa
Variable input;
Function model;
if (File.Exists(_modelFile))
{
model = Function.Load(_modelFile, DeviceDescriptor.GPUDevice(0));
input = model.Arguments.Single(a => a.Name == _featureName);
}
else
{
input = CNTKLib.InputVariable(_imageDimension, DataType.Float, _featureName);
model = BuildNetwork(input);
}
var trainer = CreateTrainer(model, labels);
IList<StreamConfiguration> streamConfigurations = new StreamConfiguration[]
{
new StreamConfiguration(_featureName, _imageSize),
new StreamConfiguration(_labelName, _classesNumber)
};
var minibatchSource = MinibatchSource.TextFormatMinibatchSource(
Path.Combine(_ressourceFolder, _trainingDataFile),
streamConfigurations,
MinibatchSource.InfinitelyRepeat);
TrainModel(minibatchSource, trainer, labels, input);
我一开始也犯了一个错误,就是使用
Variable layer model.FindByName(inputLayerName)
虽然我不得不用
Variable layer = model.Arguments.Single(a => a.Name == inputLayerName);
内容总结
以上是互联网集市为您收集整理的使用C#加载(预训练)的CNTK模型全部内容，希望文章能够帮你解决使用C#加载(预训练)的CNTK模型所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。
内容备注
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。
内容手机端
扫描二维码推送至手机访问。