public class TrainModuleUtils
extends java.lang.Object
TrainModule
Modifier and Type | Class and Description |
---|---|
static class |
TrainModuleUtils.GraphInfo |
Constructor and Description |
---|
TrainModuleUtils() |
Modifier and Type | Method and Description |
---|---|
static TrainModuleUtils.GraphInfo |
buildGraphInfo(ComputationGraphConfiguration config) |
static TrainModuleUtils.GraphInfo |
buildGraphInfo(MultiLayerConfiguration config) |
static TrainModuleUtils.GraphInfo |
buildGraphInfo(NeuralNetConfiguration config) |
public static TrainModuleUtils.GraphInfo buildGraphInfo(MultiLayerConfiguration config)
public static TrainModuleUtils.GraphInfo buildGraphInfo(ComputationGraphConfiguration config)
public static TrainModuleUtils.GraphInfo buildGraphInfo(NeuralNetConfiguration config)