Project-MONAI / MONAI

AI Toolkit for Healthcare Imaging

Home Page:https://monai.io/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add deep supervision to classification models

pooya-mohammadi opened this issue · comments

Is your feature request related to a problem? Please describe.
I have noticed deep supervision is used in nnunet and it works fine. So I think having it for classification would also be beneficial.

Describe the solution you'd like
I want classificaiton models to have an input argument like deep supervision which DynUnet has to support deep_supervision.