pytorch_pfn_extras.utils.comparer.get_default_comparer#

pytorch_pfn_extras.utils.comparer.get_default_comparer(rtol=0.0001, atol=0, equal_nan=True)#

Creates default comparer function.

The created function will compare the outputs by using torch.testing.assert_allclose with specified options.

Parameters:

  • rtol (float) – Relative tolerance.

  • atol (float) – Absolute tolerance.

  • equal_nan (bool) – If True, NaNs will be ignored.

Return type:

Callable[[str, str, str, Any, Any], None]