org.deeplearning4j.optimize.solvers.accumulation.LocalHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deeplearning4j-nn Show documentation

There is a newer version: 1.0.0-M2.1

package org.deeplearning4j.optimize.solvers.accumulation;

import lombok.NonNull;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.factory.Nd4j;

/**
 * MessageHandler implementation suited for ParallelWrapper running on single box
 *
 * PLEASE NOTE: This handler does NOT provide any network connectivity.
 *
 * @author [email protected]
 */
public class LocalHandler implements MessageHandler {
    protected transient GradientsAccumulator accumulator;

    public LocalHandler() {
        //
    }

    @Override
    public void initialize(@NonNull GradientsAccumulator accumulator) {
        this.accumulator = accumulator;
    }

    @Override
    public boolean broadcastUpdates(INDArray updates) {
        // we just loop back data immediately
        accumulator.receiveUpdate(updates);

        updates.assign(0.0);

        Nd4j.getExecutioner().commit();

        return true;
    }
}