def restore_model(folder, filename):
    """
    Loads model from an experiment folder.
    """
    path_to_ckpt = path.join(folder, "checkpoint", filename)
    with open(path_to_ckpt, 'rb') as fp:
        data = pickle.load(fp)
    return data['model']

    Loads model from an experiment folder.
    
surreal/main/rollout.py