CNTK(C#)-设置还原训练器后的最小批量大小

时间:2018-09-27 10:50:40

标签: c# neural-network deep-learning cntk

从文件中恢复培训师后,我正在寻找一种重置批次大小的方法

培训师的定义方式如下:

CNTK.Function net = CNTK.CNTKLib.ReLU(Util.Dense(net, 128, computeDevice));
net = Util.Dense(net, 1, computeDevice);
var parameterVector = new CNTK.ParameterVector((System.Collections.ICollection)net.Parameters());
var learner = CNTK.CNTKLib.AdamLearner(parameterVector, new CNTK.TrainingParameterScheduleDouble(0.001, 100), new CNTK.TrainingParameterScheduleDouble(0.9, 1), true);
trainer = CNTK.CNTKLib.CreateTrainer(net, loss_function, accuracy_function, new CNTK.LearnerVector() { learner });

并恢复:

trainer.RestoreFromCheckpoint("saved.trn");

现在如何更改小批量的大小?

0 个答案:

没有答案