tools/lightning_train_net.py · 236b15cd14af20536a8a16d2a457a6d6eb9fe8ed · OpenDAS / d2go

Set find_unused_parameters according to DDP_FIND_UNUSED_PARAMETERS · 236b15cd

Daniel Li (AI) authored Jul 07, 2021

Summary: Set find_unused_parameters according to DDP_FIND_UNUSED_PARAMETERS with DDPPlugin

Reviewed By: kazhang

Differential Revision: D29567013

fbshipit-source-id: f3ffac566a2ff046f55e692b3b24f9531913d4d4

236b15cd

lightning_train_net.py 7.35 KB

Replace lightning_train_net.py