Example usage for org.deeplearning4j.nn.conf ConvolutionMode Truncate

Introduction

In this page you can find the example usage for org.deeplearning4j.nn.conf ConvolutionMode Truncate.

Prototype

ConvolutionMode Truncate

To view the source code for org.deeplearning4j.nn.conf ConvolutionMode Truncate.

Click Source Link

Usage

From source file:weka.dl4j.layers.ConvolutionLayer.java

License:Open Source License

/**
 * Constructor for setting some defaults.
 *///from w ww  . ja  va 2 s .  com
public ConvolutionLayer() {
    setLayerName("Convolution layer");
    setActivationFunction("identity");
    setWeightInit(WeightInit.XAVIER);
    setDist(new NormalDistribution());
    setUpdater(Updater.NESTEROVS);
    setLearningRate(0.01);
    setBiasLearningRate(getLearningRate());
    setMomentum(0.9);
    setBiasInit(1.0);
    setAdamMeanDecay(0.9);
    setAdamVarDecay(0.999);
    setEpsilon(1e-6);
    setRmsDecay(0.95);
    setConvolutionMode(ConvolutionMode.Truncate);
    setKernelSize(new int[] { 5, 5 });
    setStride(new int[] { 1, 1 });
    setPadding(new int[] { 0, 0 });
    this.cudnnAlgoMode = ConvolutionLayer.AlgoMode.PREFER_FASTEST;
}

---