burn/crates/burn-import/pytorch-tests/tests
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
..
batch_norm Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
boolean Update to edition 2024 (#2931) 2025-03-20 10:28:51 -04:00
buffer Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
complex_nested Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
config Don't mention a fixed candle bug (#2689) 2025-01-13 09:10:07 -05:00
conv1d Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
conv2d Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
conv_transpose1d Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
conv_transpose2d Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
embedding Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
enum_module Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
group_norm Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
integer Update to edition 2024 (#2931) 2025-03-20 10:28:51 -04:00
key_remap Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
key_remap_chained Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
layer_norm Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
linear Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
missing_module_field [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
non_contiguous_indexes Improve test tolerance assertions (#3024) 2025-04-16 13:33:05 -04:00
top_level_key Enhance PyTorchRecorder to pass top-level key to extract state_dict (#1300) 2024-02-29 12:57:27 -06:00
test_mod.rs Refactor xtask to use tracel-xtask and refactor CI workflow (#2063) 2024-08-28 15:57:13 -04:00