public class BatchNormalizationTest
extends java.lang.Object
Modifier and Type | Field and Description |
---|---|
protected org.nd4j.linalg.api.ndarray.INDArray |
cnnEpsilon |
protected org.nd4j.linalg.api.ndarray.INDArray |
cnnInput |
protected org.nd4j.linalg.api.ndarray.INDArray |
dnnEpsilon |
protected org.nd4j.linalg.api.ndarray.INDArray |
dnnInput |
Constructor and Description |
---|
BatchNormalizationTest() |
Modifier and Type | Method and Description |
---|---|
void |
checkMeanVarianceEstimate() |
void |
checkMeanVarianceEstimateCNN() |
void |
checkSerialization() |
void |
doBefore() |
protected static Layer |
getLayer(int nOut) |
protected static Layer |
getLayer(int nOut,
double epsilon,
boolean lockGammaBeta,
double gamma,
double beta) |
void |
test2dVs4d() |
void |
testCNNBNActivationCombo() |
void |
testCnnForwardBackward() |
void |
testCnnForwardPass() |
void |
testDBNBNMultiLayer() |
void |
testDnnForwardBackward() |
void |
testDnnForwardPass() |
void |
testGradientAndUpdaters() |
protected org.nd4j.linalg.api.ndarray.INDArray dnnInput
protected org.nd4j.linalg.api.ndarray.INDArray dnnEpsilon
protected org.nd4j.linalg.api.ndarray.INDArray cnnInput
protected org.nd4j.linalg.api.ndarray.INDArray cnnEpsilon
public void doBefore()
protected static Layer getLayer(int nOut)
protected static Layer getLayer(int nOut, double epsilon, boolean lockGammaBeta, double gamma, double beta)
public void testDnnForwardPass()
public void testDnnForwardBackward()
public void testCnnForwardPass()
public void test2dVs4d()
public void testCnnForwardBackward()
public void testDBNBNMultiLayer() throws java.lang.Exception
java.lang.Exception
public void testCNNBNActivationCombo() throws java.lang.Exception
java.lang.Exception
public void checkSerialization() throws java.lang.Exception
java.lang.Exception
public void testGradientAndUpdaters() throws java.lang.Exception
java.lang.Exception
public void checkMeanVarianceEstimate() throws java.lang.Exception
java.lang.Exception
public void checkMeanVarianceEstimateCNN() throws java.lang.Exception
java.lang.Exception