@classmethod
def load(cls, path: str):
    """
        Load model state.

        :param path: state file path
        :return: state
        """
    import torch
    map_location = None
    if not torch.cuda.is_available():
        map_location = 'cpu'
    return torch.load(path, map_location=map_location)
