ray.rllib 入门实践-7：加载训练好的模型

原创

已于 2025-01-26 08:07:01 修改 · 559 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #rllib

于 2025-01-25 00:13:57 首次发布

环境配置：

torch==2.5.1

ray==2.10.0

ray[rllib]==2.10.0

ray[tune]==2.10.0

ray[serve]==2.10.0

numpy==1.23.0

python==3.9.18

直接看代码（建议先执行完 ray.rllib 入门实践-6：保存模型-优快云博客，保存了模型后再执行以下代码）

import os 
from ray.rllib.algorithms.ppo import PPO,PPOConfig
from ray.rllib.algorithms.algorithm import Algorithm 
from ray.tune.logger import pretty_print 


###################  加载 checkpoint  #####################################
checkpoint_dir = "F:/codes/RLlib_study/ray_rllib_tutorials/outputs/checkpoints"

## （推荐）方法1：用 from_chec