apple.metalperformanceshaders.enums

Class MPSNNTrainingStyle

java.lang.Object

apple.metalperformanceshaders.enums.MPSNNTrainingStyle

public final class MPSNNTrainingStyle
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
static long	CPU The weight update pass will be called in a command buffer completion callback, with a nil command buffer
static long	GPU The weight update pass will be called immediately after the gradient pass is encoded, with a nonnull command buffer
static long	None Do not train this node, for example in transfer learning

Method Summary

Methods inherited from class java.lang.Object

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

Field Detail

None

public static final long None

Do not train this node, for example in transfer learning

See Also:

Constant Field Values

CPU

public static final long CPU

The weight update pass will be called in a command buffer completion callback, with a nil command buffer

See Also:

Constant Field Values

GPU

public static final long GPU

The weight update pass will be called immediately after the gradient pass is encoded, with a nonnull command buffer

See Also:

Constant Field Values