mirror of https://github.com/tracel-ai/burn.git
![]() * 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> |
||
---|---|---|
.. | ||
batch_norm | ||
boolean | ||
buffer | ||
complex_nested | ||
config | ||
conv1d | ||
conv2d | ||
conv_transpose1d | ||
conv_transpose2d | ||
embedding | ||
enum_module | ||
group_norm | ||
integer | ||
key_remap | ||
key_remap_chained | ||
layer_norm | ||
linear | ||
missing_module_field | ||
non_contiguous_indexes | ||
top_level_key | ||
test_mod.rs |