Package apple.metalperformanceshaders.enums

Class MPSNNTrainingStyle

  • public final class MPSNNTrainingStyle
extends java.lang.Object

    • Field Summary

      Fields 
      Modifier and Type Field 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