org.deeplearning4j.nn.multilayer.MultiLayerNetwork.batchSize()方法的使用及代码示例

x33g5p2x  于2022-01-25 转载在 其他  
字(1.6k)|赞(0)|评价(0)|浏览(103)

本文整理了Java中org.deeplearning4j.nn.multilayer.MultiLayerNetwork.batchSize()方法的一些代码示例,展示了MultiLayerNetwork.batchSize()的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。MultiLayerNetwork.batchSize()方法的具体详情如下:
包路径:org.deeplearning4j.nn.multilayer.MultiLayerNetwork
类名称:MultiLayerNetwork
方法名:batchSize

MultiLayerNetwork.batchSize介绍

暂无

代码示例

代码示例来源:origin: org.deeplearning4j/deeplearning4j-ui-model

private void updateExamplesMinibatchesCounts(Model model) {
  ModelInfo modelInfo = getModelInfo(model);
  int examplesThisMinibatch = 0;
  if (model instanceof MultiLayerNetwork) {
    examplesThisMinibatch = ((MultiLayerNetwork) model).batchSize();
  } else if (model instanceof ComputationGraph) {
    examplesThisMinibatch = ((ComputationGraph) model).batchSize();
  } else if (model instanceof Layer) {
    examplesThisMinibatch = ((Layer) model).getInputMiniBatchSize();
  }
  modelInfo.examplesSinceLastReport += examplesThisMinibatch;
  modelInfo.totalExamples += examplesThisMinibatch;
  modelInfo.minibatchesSinceLastReport++;
  modelInfo.totalMinibatches++;
}

代码示例来源:origin: org.deeplearning4j/deeplearning4j-parallel-wrapper

MultiLayerNetwork workerModel = (MultiLayerNetwork) zoo[cnt].getModel();
updaters.add(workerModel.getUpdater().getStateViewArray());
batchSize += workerModel.batchSize();

代码示例来源:origin: org.deeplearning4j/deeplearning4j-parallel-wrapper_2.11

MultiLayerNetwork workerModel = (MultiLayerNetwork) zoo[cnt].getModel();
updaters.add(workerModel.getUpdater().getStateViewArray());
batchSize += workerModel.batchSize();

代码示例来源:origin: org.deeplearning4j/deeplearning4j-nn

updater.update(mln, gradAndScore.getFirst(), 0, mln.batchSize());

相关文章