s3prl_s3prl_main/s3prl/utility/allclose.py

import sys
import torch

pth1 = torch.load(sys.argv[1])
pth2 = torch.load(sys.argv[2])

print((pth1 - pth2).abs().max().item())