Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
utils.py 91 Bytes
Newer Older
Minjie Wang's avatar
[Test] Lower allclose precision (#129)  
Minjie Wang committed
1
2
3
4 
import torch as th

def allclose(a, b):
    return th.allclose(a, b, rtol=1e-4, atol=1e-4)