Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[ONNX] Reland: Update training state logic to support ScriptedModule (#…
…86745) In #86325, it was reported that ScriptedModule do not have a training attribute and will fail export because we don't expect them as input. Also - Parameterized the test_util_funs test Thanks @borisfom for the suggestion! Fixes #86325 Pull Request resolved: #86745 Approved by: https://github.com/AllenTiTaiWang, https://github.com/BowenBao
- Loading branch information
1 parent
527ebed
commit 56a744b
Showing
2 changed files
with
49 additions
and
44 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters