"docs/source/ko/tasks/summarization.md" did not exist on "77321481247787c97568c3b9f64b19e22351bab8"
Unverified Commit 0f2c7382 authored by Hugo Abonizio's avatar Hugo Abonizio Committed by GitHub
Browse files

Remove hardcoded prints in Trainer (#23432)

parent a574de30
......@@ -1105,10 +1105,10 @@ class Trainer:
for module in opt_model.modules():
if isinstance(module, nn.Embedding):
skipped += sum({p.data_ptr(): p.numel() for p in module.parameters()}.values())
print(f"skipped {module}: {skipped/2**20}M params")
logger.info(f"skipped {module}: {skipped/2**20}M params")
manager.register_module_override(module, "weight", {"optim_bits": 32})
logger.debug(f"bitsandbytes: will optimize {module} in fp32")
print(f"skipped: {skipped/2**20}M params")
logger.info(f"skipped: {skipped/2**20}M params")
if is_sagemaker_mp_enabled():
self.optimizer = smp.DistributedOptimizer(self.optimizer)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment