public class GradCheckUtil extends Object
Constructor and Description |
---|
GradCheckUtil() |
Modifier and Type | Method and Description |
---|---|
static boolean |
checkGradients(SameDiff sd,
Map<String,INDArray> placeholderValues,
boolean print,
boolean exitOnFirstFailure) |
static boolean |
checkGradients(SameDiff sd,
Map<String,INDArray> placeholderValues,
double eps,
double maxRelError,
double minAbsError,
boolean print,
boolean exitOnFirstFailure) |
static boolean |
checkGradients(SameDiff sd,
Map<String,INDArray> placeholderValues,
double eps,
double maxRelError,
double minAbsError,
boolean print,
boolean exitOnFirstFailure,
boolean skipValidation,
boolean debugMode,
Set<String> skipVariables,
Map<String,INDArray> gradCheckMask) |
static boolean |
checkGradients(SameDiff sd,
Map<String,INDArray> placeholderValues,
String... skipVariables) |
static boolean |
checkGradients(TestCase t) |
static void |
validateInternalState(SameDiff sd,
boolean generateAndCheckGradFn) |
public static boolean checkGradients(TestCase t)
public static boolean checkGradients(SameDiff sd, Map<String,INDArray> placeholderValues, String... skipVariables)
public static boolean checkGradients(SameDiff sd, Map<String,INDArray> placeholderValues, boolean print, boolean exitOnFirstFailure)
public static boolean checkGradients(SameDiff sd, Map<String,INDArray> placeholderValues, double eps, double maxRelError, double minAbsError, boolean print, boolean exitOnFirstFailure)
public static boolean checkGradients(SameDiff sd, Map<String,INDArray> placeholderValues, double eps, double maxRelError, double minAbsError, boolean print, boolean exitOnFirstFailure, boolean skipValidation, boolean debugMode, Set<String> skipVariables, Map<String,INDArray> gradCheckMask)
public static void validateInternalState(SameDiff sd, boolean generateAndCheckGradFn)
Copyright © 2019. All rights reserved.