Directional-Stimulus-Prompting icon indicating copy to clipboard operation
Directional-Stimulus-Prompting copied to clipboard

KeyError: 'summarization_with_hint'

Open gangancuicuia opened this issue 1 year ago • 0 comments

Traceback (most recent call last): File "/home/zhsx/Directional-Stimulus-Prompting/scripts/training/train_text_generation.py", line 84, in main( File "/home/zhsx/Directional-Stimulus-Prompting/scripts/training/train_text_generation.py", line 46, in main trainer = OnPolicyTrainer( File "/home/zhsx/Directional-Stimulus-Prompting/rl4lms/envs/text_generation/training_utils.py", line 149, in init self._setup() File "/home/zhsx/Directional-Stimulus-Prompting/rl4lms/envs/text_generation/training_utils.py", line 157, in _setup self._reward_fn = build_reward_fn(self._reward_config) File "/home/zhsx/Directional-Stimulus-Prompting/rl4lms/envs/text_generation/training_utils.py", line 47, in build_reward_fn reward_fn = RewardFunctionRegistry.get(reward_config["id"], File "/home/zhsx/Directional-Stimulus-Prompting/rl4lms/envs/text_generation/registry.py", line 140, in get reward_cls = cls._registry[reward_fn_id] KeyError: 'summarization_with_hint'

gangancuicuia avatar Sep 21 '24 07:09 gangancuicuia