Public
  1. Public

onnxruntime

Public

Commits

Thiago CrepaldiGitHub
4fe2ffae168
Thiago Crepaldi authored and GitHub committed 6c41ed597b0
Add custom autograd function to prevent input passthrough on ORTModule (#7694)

* Changes for investigation

* Gradient for Identity

* Keep Identity betwen YieldOp and GraphOutput

* Revert debugging changes

* Add custom autograd fn to prevent input passthrough on ORTModule

* Add comment

Co-authored-by: Sherlock Huang <bahuang@OrtTrainingDev3.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>