Commit d38c398
authored
fix(pt): use eval mode in the Python interface (#4404)
Same as #4402
<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit
- **New Features**
- Enhanced model initialization by ensuring the model is set to
evaluation mode immediately after loading, improving inference accuracy.
- **Bug Fixes**
- Corrected the control flow during model setup to prevent potential
issues during evaluation.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->
Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>1 parent cb7a0d3 commit d38c398
1 file changed
+1
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
143 | 143 | | |
144 | 144 | | |
145 | 145 | | |
| 146 | + | |
146 | 147 | | |
147 | 148 | | |
148 | 149 | | |
| |||
0 commit comments