Field Detail

• sourceModel

  protected Model sourceModel

• workers

  protected int workers

• replicas

  protected List<Model> replicas

• rootDevice

  protected boolean rootDevice

• loadBalanceMode

  protected LoadBalanceMode loadBalanceMode

• targetDeviceId

  protected int targetDeviceId

• position

  protected final AtomicLong position

• modelLock

  protected final ReentrantReadWriteLock modelLock

• queue

  protected final BlockingQueue<Model> queue

• isCG

  protected transient boolean isCG

• isMLN

  protected transient boolean isMLN

Constructor Detail

• ModelHolder

  protected ModelHolder()

Method Detail

• init

  protected void init()

• acquireModel

  protected Model acquireModel()
                        throws InterruptedException

  Throws:

  InterruptedException

• releaseModel

  protected void releaseModel(Model model)

• output

  protected INDArray[] output(INDArray[] input,
                              INDArray[] inputMasks)

• updateModel

  protected void updateModel(@NonNull
                             @NonNull Model model)