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

torchtnt.utils.oom.is_out_of_cpu_memory

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

Returns True if the exception is related to CPU OOM

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