org.nd4j.autodiff.listeners.BaseListener Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of nd4j-api Show documentation
There is a newer version: 1.0.0-M2.1
package org.nd4j.autodiff.listeners;

import org.nd4j.autodiff.listeners.records.LossCurve;
import org.nd4j.autodiff.samediff.SameDiff;
import org.nd4j.autodiff.samediff.internal.SameDiffOp;
import org.nd4j.autodiff.samediff.internal.Variable;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.dataset.api.MultiDataSet;

/**
 * A base/abstract {@link Listener} with all methods implemented as no-op.
 * Extend this for custom listeners to selectively override only the required methods
 *
 * If you want to use evaluations in your listener, use {@link BaseEvaluationListener}
 *
 * @author Alex Black
 */
public abstract class BaseListener implements Listener {


    @Override
    public ListenerVariables requiredVariables(SameDiff sd){
        return ListenerVariables.empty();
    }

    @Override
    public void epochStart(SameDiff sd, At at) {
        //No op
    }

    @Override
    public ListenerResponse epochEnd(SameDiff sd, At at, LossCurve lossCurve, long epochTimeMillis) {
        return ListenerResponse.CONTINUE;
    }

    @Override
    public ListenerResponse validationDone(SameDiff sd, At at, long validationTimeMillis) {
        //No op
        return ListenerResponse.CONTINUE;
    }

    @Override
    public void iterationStart(SameDiff sd, At at, MultiDataSet data, long etlMs) {
        //No op
    }

    @Override
    public void iterationDone(SameDiff sd, At at, MultiDataSet dataSet, Loss loss) {
        //No op
    }

    @Override
    public void operationStart(SameDiff sd, Operation op) {
        //No op
    }

    @Override
    public void operationEnd(SameDiff sd, Operation op) {
        //No op
    }

    @Override
    public void preOpExecution(SameDiff sd, At at, SameDiffOp op) {
        //No op
    }

    @Override
    public void opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, INDArray[] outputs) {
        //No op
    }

    @Override
    public void activationAvailable(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, String varName,
            INDArray activation) {
        //No op
    }

    @Override
    public void preUpdate(SameDiff sd, At at, Variable v, INDArray update) {
        //No op
    }
}