apple.metalperformanceshaders.protocol

Interface MPSNNTrainableNode

All Known Implementing Classes:

MPSCNNBatchNormalizationGradientNode, MPSCNNBatchNormalizationNode, MPSCNNBinaryConvolutionNode, MPSCNNBinaryFullyConnectedNode, MPSCNNConvolutionGradientNode, MPSCNNConvolutionNode, MPSCNNConvolutionTransposeGradientNode, MPSCNNConvolutionTransposeNode, MPSCNNFullyConnectedGradientNode, MPSCNNFullyConnectedNode, MPSCNNGroupNormalizationGradientNode, MPSCNNGroupNormalizationNode, MPSCNNInstanceNormalizationGradientNode, MPSCNNInstanceNormalizationNode

public interface MPSNNTrainableNode

Method Summary

Modifier and Type	Method and Description
void	setTrainingStyle(long value) Configure whether and when neural network training parameters are updated
long	trainingStyle() Configure whether and when neural network training parameters are updated

Method Detail

setTrainingStyle

void setTrainingStyle(long value)

Configure whether and when neural network training parameters are updated

Default: MPSNNTrainingStyleUpdateDeviceGPU

trainingStyle

long trainingStyle()

Configure whether and when neural network training parameters are updated

Default: MPSNNTrainingStyleUpdateDeviceGPU