Trainer.StepFunction

Enclosing class:: Trainer

Functional Interface:: This is a functional interface and can therefore be used as the assignment target for a lambda expression or method reference.

@FunctionalInterface public static interface Trainer.StepFunction

Method Summary

Modifier and Type

Method

Description

double

trainStep(int batchSize)

Runs one optimization step and returns the average loss for that step.

Method Details
- trainStep
  
  double trainStep(int batchSize)
  
  Runs one optimization step and returns the average loss for that step.