| 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 © 2021. All rights reserved.