Skip to content

Commit 7bf006e

Browse files
committed
[Refactor] refactor pack to support greedy, balance, native pack strategy
1 parent bedd4d4 commit 7bf006e

File tree

5 files changed

+445
-421
lines changed

5 files changed

+445
-421
lines changed

xtuner/v1/rl/base/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from .controller import TrainingController, TrainingControllerProxy, TrainingStepTimeLog
1+
from .controller import TrainingController, TrainingControllerProxy, TrainingLogInfo
22
from .loss import BaseRLLossConfig, RLLossContextInputItem
33
from .worker import (
44
TrainingWorker,
@@ -20,6 +20,6 @@
2020
"BaseRLLossConfig",
2121
"RLLossContextInputItem",
2222
"WorkerLogItem",
23-
"TrainingStepTimeLog",
2423
"WorkerInputItem",
24+
"TrainingLogInfo",
2525
]

0 commit comments

Comments
 (0)