Motivation

[众智][PyTorch]3DUNet模型更改DDP参数find_unused_parameters=True

Modification

添加DDP参数find_unused_parameters=True

Self-test (Optional)

在arm 910ProB使用torch2.0 0510的版本跑8卡性能为450.056，性能达标

BC-breaking (Optional)

If there are compatibility issues, such as dependencies on cann/torch_npu versions, they need to be explained in the PR.

Checklist

Before PR:

• The new code needs to comply with the Clean Code specification.
• The PR content is self-checked, and the expression can be clear and the writing standardized

After PR:

• CLA has been signed and all committers have signed the CLA in this PR.
• The ci-pipeline is passed, Code Check is passed.