Examples with NoUpdateStorage org.nd4j.parameterserver.updater.storage.NoUpdateStorage used on opensource projects

Example 1 with NoUpdateStorage

use of org.nd4j.parameterserver.updater.storage.NoUpdateStorage in project nd4j by deeplearning4j.

the class ParameterServerUpdaterTests method synchronousTest.

@Test
public void synchronousTest() {
    int cores = Runtime.getRuntime().availableProcessors();
    ParameterServerUpdater updater = new SynchronousParameterUpdater(new NoUpdateStorage(), new InMemoryNDArrayHolder(Nd4j.zeros(2, 2)), cores);
    for (int i = 0; i < cores; i++) {
        updater.update(NDArrayMessage.wholeArrayUpdate(Nd4j.ones(2, 2)));
    }
    assertTrue(updater.shouldReplicate());
    updater.reset();
    assertFalse(updater.shouldReplicate());
    assumeNotNull(updater.toJson());
}