首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >在ray-rllib中保存算法训练器的模型和检查点。

在ray-rllib中保存算法训练器的模型和检查点。
EN

Stack Overflow用户
提问于 2021-06-08 04:22:09
回答 1查看 138关注 0票数 0

有谁知道如何在ray-rllib中为算法训练器模型做检查点和保存模型?

我知道这对于ray.tune是可用的,但是对于rllib算法,似乎不可能直接这样做。

EN

回答 1

Stack Overflow用户

发布于 2022-05-25 15:01:43

训练器类有save_checkpoint方法和load_checkpoint方法。

代码语言:javascript
复制
 @override(Trainable)
def save_checkpoint(self, checkpoint_dir: str) -> str:
    checkpoint_path = os.path.join(
        checkpoint_dir, "checkpoint-{}".format(self.iteration)
    )
    pickle.dump(self.__getstate__(), open(checkpoint_path, "wb"))

    return checkpoint_path

@override(Trainable)
def load_checkpoint(self, checkpoint_path: str) -> None:
    extra_data = pickle.load(open(checkpoint_path, "rb"))
    self.__setstate__(extra_data)
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/67881373

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档