interface.py 文件源码

def load(cls, path: Union[str, IO]) -> 'TorchModel':
        with open_file(path, 'rb') as infile:
            state = pickle.load(infile)
        model = cls.__new__(cls)
        model.__setstate__(state)
        return model

    # for using pickle.dump/load directly