Commit 128f4bea authored by Myle Ott's avatar Myle Ott Committed by Facebook Github Bot
Browse files

Fix gating for find_unused_parameters

Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/600

Differential Revision: D15469322

Pulled By: myleott

fbshipit-source-id: fdefa8efbb10e48b2a04a6bc10404fd2f3f21ecf
parent 72a5487c
......@@ -36,7 +36,6 @@ def DistributedFairseqModel(args, model):
output_device=args.device_id,
broadcast_buffers=False,
bucket_cap_mb=args.bucket_cap_mb,
find_unused_parameters=args.find_unused_parameters
)
# Maintain backward compatibility
if 'check_reduction' in inspect.getargspec(ddp_class)[0]:
......
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