Package io.github.kirstenali.deepj.training

Interface 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
    float
    trainStep(int batchSize)
    Runs one optimization step and returns the average loss for that step.

  • Method Details

    • trainStep

      float trainStep(int batchSize)
      Runs one optimization step and returns the average loss for that step.