Examples with BandReducerLayer com.simiacryptus.mindseye.layers.cudnn.BandReducerLayer used on opensource projects

Example 1 with BandReducerLayer

use of com.simiacryptus.mindseye.layers.cudnn.BandReducerLayer in project MindsEye by SimiaCryptus.

the class VGG16_HDF5 method phase3b.

/**
 * Phase 3 b.
 */
protected void phase3b() {
    add(new SoftmaxActivationLayer().setAlgorithm(SoftmaxActivationLayer.SoftmaxAlgorithm.ACCURATE).setMode(SoftmaxActivationLayer.SoftmaxMode.CHANNEL));
    add(new BandReducerLayer().setMode(getFinalPoolingMode()));
}

Example 2 with BandReducerLayer

use of com.simiacryptus.mindseye.layers.cudnn.BandReducerLayer in project MindsEye by SimiaCryptus.

the class VGG19_HDF5 method phase3b.

/**
 * Phase 3 b.
 */
protected void phase3b() {
    add(new SoftmaxActivationLayer().setAlgorithm(SoftmaxActivationLayer.SoftmaxAlgorithm.ACCURATE).setMode(SoftmaxActivationLayer.SoftmaxMode.CHANNEL));
    add(new BandReducerLayer().setMode(getFinalPoolingMode()));
}