ai.djl.pytorch.engine

Class PtGradientCollector

java.lang.Object

ai.djl.pytorch.engine.PtGradientCollector

All Implemented Interfaces:

ai.djl.training.GradientCollector, java.lang.AutoCloseable

public class PtGradientCollector
extends java.lang.Object
implements ai.djl.training.GradientCollector

PtGradientCollector is the PyTorch implementation of GradientCollector.

Constructor Summary

Constructors

Constructor and Description
PtGradientCollector()

Method Summary

Modifier and Type	Method and Description
void	backward(ai.djl.ndarray.NDArray target)
void	close()

Methods inherited from class java.lang.Object

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

Constructor Detail

PtGradientCollector

public PtGradientCollector()

Method Detail

backward

public void backward(ai.djl.ndarray.NDArray target)

Specified by:

backward in interface ai.djl.training.GradientCollector

close

public void close()

Specified by:

close in interface ai.djl.training.GradientCollector

Specified by:

close in interface java.lang.AutoCloseable