public interface Trainable
Modifier and Type | Method and Description |
---|---|
TrainingConfig |
getConfig() |
INDArray |
getGradientsViewArray() |
int |
numParams() |
INDArray |
params() |
Map<String,INDArray> |
paramTable(boolean backpropOnly) |
TrainingConfig getConfig()
int numParams()
INDArray params()
Map<String,INDArray> paramTable(boolean backpropOnly)
backpropOnly
- If true: return only parameters that are not exclusively used for layerwise pretrainingINDArray getGradientsViewArray()
Copyright © 2018. All rights reserved.