问题遇到的现象和发生背景
在调用stable_baseline3包跑PPO的时候报这个错,具体错误处是在
用代码块功能插入代码,请勿粘贴截图
#导入包
from stable_baselines3 import PPO
from stable_baselines3.common.vec_env import SubprocVecEnv, DummyVecEnv
from stable_baselines3.common.callbacks import (
EvalCallback,
StopTrainingOnRewardThreshold,
)
from stable_baselines3.common.policies import ActorCriticPolicy, BasePolicy
#实例化设定停止训练的阈值
# threshold settings for training curriculum
# type can be either 'succ' or 'rew'
trainstage_cb = InitiateNewTrainStage(
n_envs=args.n_envs,
treshhold_type="succ",
upper_threshold=0.9,
lower_threshold=0.7,
task_mode=params["task_mode"],
verbose=1,
)
# stop training on reward threshold callback
stoptraining_cb = StopTrainingOnRewardThreshold(
treshhold_type="succ", threshold=0.95, verbose=1
)
运行结果及报错内容
Traceback (most recent call last):
File "scripts/training/train_agent.py", line 217, in <module>
main()
File "scripts/training/train_agent.py", line 80, in main
stoptraining_cb = StopTrainingOnRewardThreshold(
TypeError: __init__() got an unexpected keyword argument 'treshhold_type'