burn/crates/burn-import/pytorch-tests/tests/complex_nested
Guillaume Lagrange 17e3167a70
Improve test tolerance assertions (#3024)
* wip new tolerance

* use the new assert_approx_eq

* fix or tag broken tests

* Fix more tests

* All tests pass cuda/wgpu

* Format

* Fix other modules tests tolerance

* Fix onnx import tolerances

* Remove broken comments

* Expected values are not defined with enough precision anyway

* Less strict tolerance for tch

* Fix docstring

* More tolerance

* Revert to 1e-1 abs (previous tolerance, probably set for tch)

* Tolerances againnnn

* Fix review

* relax some tolerances

---------

Co-authored-by: maxtremblay <t.maxime@pm.me>
2025-04-16 13:33:05 -04:00
..
complex_nested.pt [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
export_weights.py [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
mod.rs Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00