Progress indicator icon still displayed in napari after getting OOM error

[marabuuu]

Hi all,

I was trying to make a prediction with Custom Model 2D/3D and I was getting the following out of memory error in the terminal:

Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. This isn't available when running in Eager mode.
 [Op:__inference_predict_function_2519]

but in napari, the progress indicator icon was still displayed (see yellow box in screenshot):

I think it would be a good idea to disable this Progress indicator icon when an error occurs so the users who are not looking into their terminal window do not think the model is still predicting and are not waiting for output.

Best,
Mara

[uschmidt83]

Hi @marabuuu, thanks for the suggestion. I will probably fix this in the next release, but I don't yet know when that will be.

Sorry for the late reply!

Best,
Uwe