Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix(pt): detach computed descriptor tensor to prevent OOM (deepmodeli…
…ng#4547) Fix deepmodeling#4544. <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **Optimization** - Improved descriptor handling in the atomic model to enhance computational efficiency and memory management during backpropagation. <!-- end of auto-generated comment: release notes by coderabbit.ai --> Signed-off-by: Jinzhe Zeng <[email protected]>
- Loading branch information