Patch mmcv imports for v2.x / mmengine compatibility

cavp/model/cavp_modules.py

@@ -3,16 +3,20 @@ import warnings
 import torch
 import torch.nn as nn
 from mmcv.cnn import ConvModule, kaiming_init
-from mmcv.runner import _load_checkpoint, load_checkpoint
-from mmcv.utils import print_log
+from mmengine.runner import load_checkpoint
+try:
+        from mmengine.runner.checkpoint import _load_checkpoint
+except ImportError:
+        _load_checkpoint = load_checkpoint
+from mmengine.logging import print_log
 
 import warnings
 import torch.nn as nn
 import torch.utils.checkpoint as cp
 from mmcv.cnn import (ConvModule, NonLocal3d, build_activation_layer,
                       constant_init, kaiming_init)
-from mmcv.runner import _load_checkpoint, load_checkpoint
-from mmcv.utils import _BatchNorm
+from mmengine.runner import load_checkpoint
+from torch.nn.modules.batchnorm import _BatchNorm
 from torch.nn.modules.utils import _ntuple, _triple
 
 from itertools import repeat
@@ -23,6 +27,10 @@ from torch.utils.checkpoint import checkpoint
 from torch.nn import functional as F
 from collections import OrderedDict
 
+import logging
+def get_root_logger(log_level=logging.DEBUG):
+        return logging.getLogger('mmengine')
+
 class BasicBlock3d(nn.Module):
     """BasicBlock 3d block for ResNet3D.
     Args: