Package ai.djl.pytorch.engine

Class PtGradientCollector

  • All Implemented Interfaces:
    ai.djl.training.GradientCollector, java.lang.AutoCloseable
    public final class PtGradientCollector
extends java.lang.Object
implements ai.djl.training.GradientCollector
    PtGradientCollector is the PyTorch implementation of GradientCollector.

    • Constructor Summary

      Constructors 
      Constructor Description
      PtGradientCollector()
      Constructs a new PtGradientCollector instance.

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      void backward​(ai.djl.ndarray.NDArray target)
      void close()
      void zeroGradients()

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

    • Constructor Detail

      • PtGradientCollector

        public PtGradientCollector()
        Constructs a new PtGradientCollector instance.

    • Method Detail

      • backward

        public void backward​(ai.djl.ndarray.NDArray target)
        Specified by:
        backward in interface ai.djl.training.GradientCollector

      • zeroGradients

        public void zeroGradients()
        Specified by:
        zeroGradients in interface ai.djl.training.GradientCollector

      • close

        public void close()
        Specified by:
        close in interface java.lang.AutoCloseable
        Specified by:
        close in interface ai.djl.training.GradientCollector