tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

don't use include_tokens_per_second for GRPO (#2931) [skip ci]

Browse Source 
* don't use include_tokens_per_second for GRPO

* use blocklist instead
This commit is contained in:
Wing Lian
2025-07-17 15:31:21 -04:00
committed by GitHub
parent 9f2bb188a4
commit 8e41317250
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/core/trainers/grpo/__init__.py
View File

@@ -148,7 +148,7 @@ class GRPOStrategy:
@classmethod @classmethod
def get_blocklist_args_kwargs(cls) -> list[str]: def get_blocklist_args_kwargs(cls) -> list[str]:
return ["dataset_num_proc", "max_length"] return ["dataset_num_proc", "max_length", "include_tokens_per_second"]
@classmethod @classmethod
def get_reward_func(cls, reward_func_fqn: str) -> RewardFunc: def get_reward_func(cls, reward_func_fqn: str) -> RewardFunc: