Page MenuHomePhabricator

Model selection for GetTaskModelOperator
Closed, ResolvedPublic

Description

In one of my dags I use NnUnetOperator and to pass which task I want to use I use task_id argument in GetTaskModelOperator. However, the GetTaskModelOperator then checks if I’ve got all the models (e.g., 2d, 3d_lowres, etc.), and since I’ve got only 3d_fullres, reports an error. How can I make it use only 3d_fullres?

Event Timeline

t543a triaged this task as Wishlist priority.Feb 3 2022, 10:00 AM
t543a created this task.