Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "3fcfbe7549d9694f96e1f19630add4adf99dd421"
Unverified Commit a5392ee7 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix test for torchneuroncore in Trainer (#22028)

parent de81adf9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/trainer.py
View file @ a5392ee7
...@@ -1538,7 +1538,7 @@ class Trainer: ...@@ -1538,7 +1538,7 @@ class Trainer:
if self.args.ddp_bucket_cap_mb is not None: if self.args.ddp_bucket_cap_mb is not None:
kwargs["bucket_cap_mb"] = self.args.ddp_bucket_cap_mb kwargs["bucket_cap_mb"] = self.args.ddp_bucket_cap_mb
if is_torch_neuroncore_available: if is_torch_neuroncore_available():
return model return model
model = nn.parallel.DistributedDataParallel( model = nn.parallel.DistributedDataParallel(
model, model,
......
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