org.deeplearning4j.nn.conf.layers

Class LocalResponseNormalization.Builder

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer.Builder<LocalResponseNormalization.Builder>

org.deeplearning4j.nn.conf.layers.LocalResponseNormalization.Builder

Enclosing class:

LocalResponseNormalization

public static class LocalResponseNormalization.Builder
extends Layer.Builder<LocalResponseNormalization.Builder>

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	cudnnAllowFallback When using CuDNN and an error is encountered, should fallback to the non-CuDNN implementatation be allowed? If set to false, an exception in CuDNN will be propagated back to the user.

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

Constructor Summary

Constructors

Constructor and Description
Builder()
Builder(double k, double alpha, double beta)
Builder(double k, double n, double alpha, double beta)

Method Summary

Modifier and Type	Method and Description
LocalResponseNormalization.Builder	alpha(double alpha) LRN scaling constant alpha.
LocalResponseNormalization.Builder	beta(double beta) Scaling constant beta.
LocalResponseNormalization	build()
LocalResponseNormalization.Builder	cudnnAllowFallback(boolean allowFallback) When using CuDNN and an error is encountered, should fallback to the non-CuDNN implementatation be allowed? If set to false, an exception in CuDNN will be propagated back to the user.
LocalResponseNormalization.Builder	k(double k) LRN scaling constant k.
LocalResponseNormalization.Builder	n(double n) Number of adjacent kernel maps to use when doing LRN.

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

cudnnAllowFallback

protected boolean cudnnAllowFallback

When using CuDNN and an error is encountered, should fallback to the non-CuDNN implementatation be allowed? If set to false, an exception in CuDNN will be propagated back to the user. If false, the built-in (non-CuDNN) implementation for BatchNormalization will be used

Constructor Detail

Builder

public Builder(double k,
               double n,
               double alpha,
               double beta)

Builder

public Builder(double k,
               double alpha,
               double beta)

Builder

public Builder()

Method Detail

k

public LocalResponseNormalization.Builder k(double k)

LRN scaling constant k. Default: 2

Parameters:

k - Scaling constant

n

public LocalResponseNormalization.Builder n(double n)

Number of adjacent kernel maps to use when doing LRN. default: 5

Parameters:

n - Number of adjacent kernel maps

alpha

public LocalResponseNormalization.Builder alpha(double alpha)

LRN scaling constant alpha. Default: 1e-4

Parameters:

alpha - Scaling constant

beta

public LocalResponseNormalization.Builder beta(double beta)

Scaling constant beta. Default: 0.75

Parameters:

beta - Scaling constant

cudnnAllowFallback

public LocalResponseNormalization.Builder cudnnAllowFallback(boolean allowFallback)

When using CuDNN and an error is encountered, should fallback to the non-CuDNN implementatation be allowed? If set to false, an exception in CuDNN will be propagated back to the user. If false, the built-in (non-CuDNN) implementation for BatchNormalization will be used

Parameters:

allowFallback - Whether fallback to non-CuDNN implementation should be used

build

public LocalResponseNormalization build()

Specified by:

build in class Layer.Builder<LocalResponseNormalization.Builder>