Search in sources :

Example 1 with SoftmaxActivationLayer

use of com.simiacryptus.mindseye.layers.cudnn.SoftmaxActivationLayer 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()));
}
Also used : SoftmaxActivationLayer(com.simiacryptus.mindseye.layers.cudnn.SoftmaxActivationLayer) BandReducerLayer(com.simiacryptus.mindseye.layers.cudnn.BandReducerLayer)

Example 2 with SoftmaxActivationLayer

use of com.simiacryptus.mindseye.layers.cudnn.SoftmaxActivationLayer 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()));
}
Also used : SoftmaxActivationLayer(com.simiacryptus.mindseye.layers.cudnn.SoftmaxActivationLayer) BandReducerLayer(com.simiacryptus.mindseye.layers.cudnn.BandReducerLayer)

Aggregations

BandReducerLayer (com.simiacryptus.mindseye.layers.cudnn.BandReducerLayer)2 SoftmaxActivationLayer (com.simiacryptus.mindseye.layers.cudnn.SoftmaxActivationLayer)2