Package | Description |
---|---|
org.nd4j.autodiff.listeners.impl | |
org.nd4j.autodiff.samediff | |
org.nd4j.autodiff.samediff.internal |
Constructor and Description |
---|
HistoryListener(TrainingConfig tc) |
Modifier and Type | Method and Description |
---|---|
TrainingConfig |
TrainingConfig.Builder.build() |
static TrainingConfig |
TrainingConfig.fromJson(@NonNull String json) |
Modifier and Type | Method and Description |
---|---|
void |
SameDiff.setTrainingConfig(TrainingConfig trainingConfig)
Set the training configuration (
TrainingConfig ) for the SameDiff instance. |
Modifier and Type | Field and Description |
---|---|
protected TrainingConfig |
TrainingSession.config |
Modifier and Type | Method and Description |
---|---|
Loss |
TrainingSession.trainingIteration(TrainingConfig config,
Map<String,INDArray> placeholders,
Set<String> paramsToTrain,
Map<String,GradientUpdater> updaters,
MultiDataSet batch,
List<String> lossVariables,
List<Listener> listeners,
At at)
Perform one iteration of training - i.e., do forward and backward passes, and update the parameters
|
Copyright © 2020. All rights reserved.