louis_lifu
/
AReaL
mirror of https://github.com/inclusionAI/AReaL
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

.

This commit is contained in:
bowei.fw 2025-07-04 16:28:32 +08:00
parent 3a8796b93a
commit 89a8d8c46a
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
realhf/impl/model/interface/ppo_interface.py
View File

@ -207,8 +207,7 @@ def splited_sum_bool_tensor(t: torch.BoolTensor, chunk_size=256 * 1024 * 1024) -
@dataclasses.dataclass
class PPOActor(model_api.ModelInterface):
engine: xxx
class PPOActorInterface(model_api.ModelInterface):
n_minibatches: int = 4
# Use dict here to allow argument passing through commandline.