diff --git a/mim/utils/default.py b/mim/utils/default.py index f26c5a0..2e53299 100644 --- a/mim/utils/default.py +++ b/mim/utils/default.py @@ -23,6 +23,9 @@ PKG2PROJECT = { 'mmedit': 'mmediting', 'mmocr': 'mmocr', 'mmgen': 'mmgeneration', + 'mmselfsup': 'mmselfsup', + 'mmrotate': 'mmrotate', + 'mmflow': 'mmflow', } # TODO: Should directly infer MODULE name from PKG info PKG2MODULE = {