AsDiscreted extremely slow

[sunshichen]

I tried to test monailabel follow the official guidline on a remote server and use 3Dslicer as annotation client. How ever it seems unusual that AsDiscreted in post-processing is extremely slow, the log as below:

I used deepedit_multilabel sample on Ubuntu18.04 with Nvidia GTX 1080Ti.

[Nic-Ma]

Hi @sunshichen ,

Could you please help provide the full test program? Then @SachidanandAlle maybe can help figure out more ideas.

Thanks in advance.

[rijobro]

Duplicate of Project-MONAI/MONAILabel#604.

[SachidanandAlle]

The quick fix for this shall be.. use EnsureTyped(keys="pred", device="cuda") as the first post transform

You can refer: https://github.com/Project-MONAI/MONAILabel/blob/main/sample-apps/deepedit_multilabel/lib/infer.py#L97

This can help to speed up the post transform using GPU..