def METHOD_NAME(path, *args, **kwargs):
    import torch
    state = torch.load(path, map_location="cpu")
    for key in list(state.keys()):
        v = state.pop(key)
        state[key] = v.numpy()
    return state