mask_soft_teacher is not in the models registry'

[alaa-shubbak]

Hello , I am working on mask rcnn model , and follow some of your instruction to make the training woks well.

but unfortunately , I got an error : is not in the models registry'

my environment is as bellow:

I follow this https://github.com/microsoft/SoftTeacher/issues/154 and this https://github.com/microsoft/SoftTeacher/issues/146

also i try to reinstall ssod using python setup.py but i still get the error.

I try to add the module in the registry by using this command :

custom_imports = dict(
    imports=['ssod.models.mask_soft_teacher'],
    allow_failed_imports=False)

I wrote it in my config as following :

and that also produces error as bellow:

any solution please