Skip to content

Commit 1aa91be

Browse files
authored
clean print in auto_trainer (#9357)
1 parent cd22b0d commit 1aa91be

1 file changed

Lines changed: 0 additions & 2 deletions

File tree

paddlenlp/trainer/auto_trainer.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,9 +141,7 @@ def _wrap_for_auto(self, model, train_dataloader):
141141
if self.enable_autocast_context_manager:
142142
unified_strategy.amp.custom_black_list.extend(["reduce_sum", "c_softmax_with_cross_entropy"])
143143
if self.args.fp16_opt_level == "O2":
144-
print("custom_white_list", unified_strategy.amp.custom_white_list, flush=1)
145144
unified_strategy.amp.custom_white_list.extend(["lookup_table", "lookup_table_v2"])
146-
print("custom_white_list", unified_strategy.amp.custom_white_list, flush=1)
147145

148146
# dist.to_static() obtains the input spec information through next(dataloader), but this has side effects
149147
# on the passed-in dataloader, altering the state of the sampler of the dataloader. In some cases, once

0 commit comments

Comments
 (0)