This can check if the state_dict from 2 pytorch models are the same

both-torch-models-equal.py

def is_state_dict_equal(dict1, dict2):
    import torch
    for key in dict1:
        if key not in dict2:
            print(f"Key {key} not in second dict")
            return False
        if not torch.all(torch.eq(dict1[key], dict2[key])):
            print(f"Difference in values for key {key}")
            return False
    return True