recover logging

Former-commit-id: d74014496e4ccda2de4482075a91747854facddd
2023-06-06 21:36:37 +08:00
parent bf5ad34196
commit 12094c1db5
6 changed files with 17 additions and 16 deletions
--- a/src/utils/ppo.py
+++ b/src/utils/ppo.py
@@ -16,12 +16,12 @@ from .config import FinetuningArguments

 from .other import (
    AverageMeter,
-    get_main_logger,
+    get_logger,
    get_logits_processor
 )


-logger = get_main_logger(__name__)
+logger = get_logger(__name__)


 def replace_model(model: AutoModelForCausalLMWithValueHead, target: Literal["default", "reward"]) -> None: