Register compose

open-mmlab · Feb 14, 2023 · a7ad6d8 · a7ad6d8
1 parent 834f94d
commit a7ad6d8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmcv/transforms/wrappers.py b/mmcv/transforms/wrappers.py
@@ -36,6 +36,7 @@ def nullcontext(resource=None):
             pass
 
 
+@TRANSFORMS.register_module()
 class Compose(BaseTransform):
     """Compose multiple transforms sequentially.