Python prepare_for_backwardの例

コード例 #1

 def pre_backward(self, closure_loss: torch.Tensor) -> None:
     """Run before precision plugin executes backward."""
     if not self.lightning_module.automatic_optimization:
         prepare_for_backward(self.model, closure_loss)

コード例 #2

ファイル: ddp.py プロジェクト: mauvilsa/pytorch-lightning

 def pre_backward(self, closure_loss: torch.Tensor, should_accumulate: bool,
                  optimizer: Optimizer, opt_idx: int):
     """Run before precision plugin executes backward"""
     if not self.lightning_module.automatic_optimization:
         prepare_for_backward(self.model, closure_loss)

コード例 #3

 def on_before_manual_backward(self, model: DistributedDataParallel,
                               output: Any):
     prepare_for_backward(model, output)