[NNVM][TEST] Test against numerical grad (#1505)
* [NNVM][TEST] Numerical gradient testing * [NNVM][TEST] Make some tests a little faster * Fix the failing test_top_level3 * Target exclusion for the check_function * Try to ignore singularities * grad_input_vars now can't contain shapes * Don't pass unnecessary grad_input_vars to check_function * Multiple outputs; fixes; testing of check_function * Use numerical_grads_params to pass parameters to numgrad checker * Fail when no action is requested excplicitly * Pass additional params to functions * Silence the linter issue * Simplified numgrad checking * Improved docs for check_function * Fixed the error message when no dtype is provided * Several fixes * Tests with shape/dtype inference for inputs * Don't check dense's grads on cuda * Raise an error if output dtypes haven't been inferred * Moved shape/dtype inference into a separate function; use float32 as fallback * Remove redundant dtype=float32 * Fix multiple outputs * Use check_function in the rest of the test_top_level1
Showing
- docs/api/python/nnvm/index.rst 1 addition, 0 deletionsdocs/api/python/nnvm/index.rst
- docs/api/python/nnvm/testing.rst 14 additions, 0 deletionsdocs/api/python/nnvm/testing.rst
- docs/api/python/nnvm/testing_new_ops.rst 135 additions, 0 deletionsdocs/api/python/nnvm/testing_new_ops.rst
- nnvm/python/nnvm/testing/__init__.py 1 addition, 0 deletionsnnvm/python/nnvm/testing/__init__.py
- nnvm/python/nnvm/testing/check_computation.py 641 additions, 0 deletionsnnvm/python/nnvm/testing/check_computation.py
- nnvm/tests/python/compiler/test_top_level1.py 252 additions, 216 deletionsnnvm/tests/python/compiler/test_top_level1.py
- nnvm/tests/python/compiler/test_top_level3.py 4 additions, 5 deletionsnnvm/tests/python/compiler/test_top_level3.py
- nnvm/tests/python/compiler/test_top_level4.py 47 additions, 106 deletionsnnvm/tests/python/compiler/test_top_level4.py
Loading
Please register or sign in to comment