You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

ImportError:无法从main.training.utils导入RewardLoggerCallback求助

问题分析与解决办法

核心问题

你的utils.py代码里存在运行时错误,导致模块无法正常加载,所以Python找不到RewardLoggerCallback类。

具体来说:
在on_episode_start方法中,你把episode.user_data['MainRew']初始化为整数0,但后续on_episode_step里又对它调用append()方法——整数类型根本没有append方法,这会触发AttributeError异常,直接导致模块加载失败,进而让导入语句报错。

修复步骤

  1. 修正utils.py的初始化逻辑
    把on_episode_start里的初始化代码改成空列表,才能支持后续的append操作:
episode.user_data = {
    'MainRew': []
}
  1. 修正__init__.py的导入语句
    你之前写的from main.training.utils import RewardLoggerCallback属于循环导入(在main.training包的__init__.py里导入自身包的模块),改成相对导入更稳妥:
from .utils import RewardLoggerCallback
  1. 确保Python能找到main包
    运行脚本时,要保证工作目录是main文件夹的父目录;如果无法切换目录,也可以把该父目录添加到PYTHONPATH环境变量中,比如终端执行:
export PYTHONPATH="/path/to/main_parent_dir:$PYTHONPATH"

做完以上三步,再执行from main.training.utils import RewardLoggerCallback就能正常导入了。

内容的提问来源于stack exchange,提问作者Victor M

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.16 09:13:11