remove class check in registry; add type assert in weight_init

2020-08-08 07:08:23 +08:00
parent f7843de45d
commit 206d9343cd
2 changed files with 5 additions and 3 deletions
--- a/util/registry.py
+++ b/util/registry.py
@@ -119,9 +119,9 @@ class Registry(_Registry):
        return self._module_dict.get(key, None)

    def _register_module(self, module_class, module_name=None, force=False):
-        if not inspect.isclass(module_class):
-            raise TypeError('module must be a class, '
-                            f'but got {type(module_class)}')
+        # if not inspect.isclass(module_class):
+        #     raise TypeError('module must be a class, '
+        #                     f'but got {type(module_class)}')

        if module_name is None:
            module_name = module_class.__name__