Spaces:
Running on Zero
Running on Zero
Commit ·
b267be8
1
Parent(s): 2a04a27
Fix mmcv v2 import: move kaiming_init/constant_init to mmengine.model
Browse files
cavp/model/cavp_modules.py
CHANGED
|
@@ -2,19 +2,19 @@ import warnings
|
|
| 2 |
|
| 3 |
import torch
|
| 4 |
import torch.nn as nn
|
| 5 |
-
from mmcv.cnn import ConvModule
|
| 6 |
from mmengine.runner import load_checkpoint
|
| 7 |
try:
|
| 8 |
from mmengine.runner.checkpoint import _load_checkpoint
|
| 9 |
except ImportError:
|
| 10 |
_load_checkpoint = load_checkpoint
|
| 11 |
from mmengine.logging import print_log
|
|
|
|
| 12 |
|
| 13 |
import warnings
|
| 14 |
import torch.nn as nn
|
| 15 |
import torch.utils.checkpoint as cp
|
| 16 |
-
from mmcv.cnn import (ConvModule, NonLocal3d, build_activation_layer
|
| 17 |
-
constant_init, kaiming_init)
|
| 18 |
from mmengine.runner import load_checkpoint
|
| 19 |
from torch.nn.modules.batchnorm import _BatchNorm
|
| 20 |
from torch.nn.modules.utils import _ntuple, _triple
|
|
|
|
| 2 |
|
| 3 |
import torch
|
| 4 |
import torch.nn as nn
|
| 5 |
+
from mmcv.cnn import ConvModule
|
| 6 |
from mmengine.runner import load_checkpoint
|
| 7 |
try:
|
| 8 |
from mmengine.runner.checkpoint import _load_checkpoint
|
| 9 |
except ImportError:
|
| 10 |
_load_checkpoint = load_checkpoint
|
| 11 |
from mmengine.logging import print_log
|
| 12 |
+
from mmengine.model import kaiming_init, constant_init
|
| 13 |
|
| 14 |
import warnings
|
| 15 |
import torch.nn as nn
|
| 16 |
import torch.utils.checkpoint as cp
|
| 17 |
+
from mmcv.cnn import (ConvModule, NonLocal3d, build_activation_layer)
|
|
|
|
| 18 |
from mmengine.runner import load_checkpoint
|
| 19 |
from torch.nn.modules.batchnorm import _BatchNorm
|
| 20 |
from torch.nn.modules.utils import _ntuple, _triple
|