mirror of https://github.com/tracel-ai/burn.git
17e3167a70
* 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> |
||
|---|---|---|
| .. | ||
| export_weights.py | ||
| mod.rs | ||
| non_contiguous_indexes.pt | ||