Closed njzjz closed 3 months ago
deepmd/pt/utils/auto_batch_size.py (4)
`60-60`: Enhanced detection of out-of-memory errors. Consider adding logging before clearing the cache to help with debugging and monitoring. --- `60-60`: Complex batch processing logic appears robust and well-handled. --- `60-60`: Standard method for checking GPU availability. --- `60-60`: Constructor uses sensible defaults and proper delegation.
All modified and coverable lines are covered by tests :white_check_mark:
Project coverage is 82.66%. Comparing base (
674bad7
) to head (325d32c
). Report is 116 commits behind head on devel.
:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.
I just received another error message that reports out of memory. It's a bad design of PyTorch that all errors use a general
RuntimeError
.Summary by CodeRabbit