File tree Expand file tree Collapse file tree 2 files changed +0
-2
lines changed
pytorch_lightning/plugins/training_type Expand file tree Collapse file tree 2 files changed +0
-2
lines changed Original file line number Diff line number Diff line change @@ -73,7 +73,6 @@ def _setup_models_and_optimizers(
7373
7474 optimizers = self ._wrap_optimizers (optimizers )
7575 model = ShardedDataParallel (models [0 ], sharded_optimizer = optimizers , ** self ._ddp_kwargs )
76- setattr (model , "require_backward_grad_sync" , False ) # TODO: needed?
7776 return [model ], optimizers
7877
7978 def _reinit_optimizers_with_oss (self , optimizers : List [Union [Optimizer , LightningOptimizer ]]) -> List ["OSS" ]:
Original file line number Diff line number Diff line change @@ -64,7 +64,6 @@ def _setup_models_and_optimizers(
6464
6565 optimizers = self ._wrap_optimizers (optimizers )
6666 model = ShardedDataParallel (models [0 ], sharded_optimizer = optimizers , ** self ._ddp_kwargs )
67- setattr (model , "require_backward_grad_sync" , False ) # TODO: needed?
6867 return [model ], optimizers
6968
7069 def _reinit_optimizers_with_oss (self , optimizers : List [Optimizer ]) -> List ["OSS" ]:
You can’t perform that action at this time.
0 commit comments