Function _torch_load_compat

tests/utils.py:13–18

(filepath)

def _torch_load_compat(filepath):
    kwargs = {"map_location": "cpu"}
    try:
        return torch.load(filepath, weights_only=False, **kwargs)
    except TypeError:
        return torch.load(filepath, **kwargs)


def gen_sequence(dim, max_len, sample_size):