• Docs >
  • Utils >
  • torchtnt.utils.oom.is_out_of_memory_error
Shortcuts

torchtnt.utils.oom.is_out_of_memory_error

torchtnt.utils.oom.is_out_of_memory_error(exception: BaseException) bool

Returns True if an exception is due to an OOM based on error message

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources