ai.djl.mxnet.engine

Class MxTrainer

java.lang.Object

ai.djl.mxnet.engine.MxTrainer

All Implemented Interfaces:

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

public class MxTrainer
extends java.lang.Object
implements ai.djl.training.Trainer

MxTrainer is the MXNet implementation of the Trainer.

Method Summary

Modifier and Type	Method and Description
void	close()
void	endEpoch()
protected void	finalize()
ai.djl.ndarray.NDList	forward(ai.djl.ndarray.NDList input)
java.util.List<ai.djl.Device>	getDevices()
<T extends ai.djl.training.evaluator.Evaluator> T	getEvaluator(java.lang.Class<T> clazz)
java.util.List<ai.djl.training.evaluator.Evaluator>	getEvaluators()
ai.djl.training.loss.Loss	getLoss()
ai.djl.ndarray.NDManager	getManager()
ai.djl.metric.Metrics	getMetrics()
ai.djl.Model	getModel()
void	initialize(ai.djl.ndarray.types.Shape... shapes)
ai.djl.training.GradientCollector	newGradientCollector()
void	setMetrics(ai.djl.metric.Metrics metrics)
void	step()
void	trainBatch(ai.djl.training.dataset.Batch batch)
void	validateBatch(ai.djl.training.dataset.Batch batch)

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.training.Trainer

iterateDataset

Method Detail

initialize

public void initialize(ai.djl.ndarray.types.Shape... shapes)

Specified by:

initialize in interface ai.djl.training.Trainer

newGradientCollector

public ai.djl.training.GradientCollector newGradientCollector()

Specified by:

newGradientCollector in interface ai.djl.training.Trainer

trainBatch

public void trainBatch(ai.djl.training.dataset.Batch batch)

Specified by:

trainBatch in interface ai.djl.training.Trainer

forward

public ai.djl.ndarray.NDList forward(ai.djl.ndarray.NDList input)

Specified by:

forward in interface ai.djl.training.Trainer

validateBatch

public void validateBatch(ai.djl.training.dataset.Batch batch)

Specified by:

validateBatch in interface ai.djl.training.Trainer

step

public void step()

Specified by:

step in interface ai.djl.training.Trainer

getMetrics

public ai.djl.metric.Metrics getMetrics()

Specified by:

getMetrics in interface ai.djl.training.Trainer

setMetrics

public void setMetrics(ai.djl.metric.Metrics metrics)

Specified by:

setMetrics in interface ai.djl.training.Trainer

getDevices

public java.util.List<ai.djl.Device> getDevices()

Specified by:

getDevices in interface ai.djl.training.Trainer

endEpoch

public void endEpoch()

Specified by:

endEpoch in interface ai.djl.training.Trainer

getLoss

public ai.djl.training.loss.Loss getLoss()

Specified by:

getLoss in interface ai.djl.training.Trainer

getModel

public ai.djl.Model getModel()

Specified by:

getModel in interface ai.djl.training.Trainer

getEvaluators

public java.util.List<ai.djl.training.evaluator.Evaluator> getEvaluators()

Specified by:

getEvaluators in interface ai.djl.training.Trainer

getEvaluator

public final <T extends ai.djl.training.evaluator.Evaluator> T getEvaluator(java.lang.Class<T> clazz)

Specified by:

getEvaluator in interface ai.djl.training.Trainer

getManager

public ai.djl.ndarray.NDManager getManager()

Specified by:

getManager in interface ai.djl.training.Trainer

finalize

protected void finalize()
                 throws java.lang.Throwable

Overrides:

finalize in class java.lang.Object

Throws:

java.lang.Throwable

close

public void close()

Specified by:

close in interface ai.djl.training.Trainer

Specified by:

close in interface java.lang.AutoCloseable