NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.17k stars 1.35k forks source link

Cleanup usage of `self.assertTrue(torch.allclose(...` #1742

Closed eqy closed 8 months ago

eqy commented 8 months ago

Cleanup so that we have better signal (i.e., are numerical mismatches small or large?) in logs of failing tests.

Some files appear to be mangled because they were in Windows? format somehow? I've converted them to unix in this PR.

CC @crcrpar @Fuzzkatt