Merge pull request #3 from Clarifai/796-3

fix
Clarifai · Oct 1, 2024 · a5a9fe3 · a5a9fe3
2 parents c78b72a + efb5851
commit a5a9fe3
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/mmcls/apis/test.py b/mmcls/apis/test.py
@@ -10,7 +10,7 @@
 import torch
 import torch.distributed as dist
 from mmcv.image import tensor2imgs
-from mmengine.runner import get_dist_info
+from mmengine.dist.utils import get_dist_info
 
 
 def single_gpu_test(model,

diff --git a/mmcls/datasets/builder.py b/mmcls/datasets/builder.py
@@ -7,7 +7,7 @@
 import numpy as np
 import torch
 from mmcv.parallel import collate
-from mmengine.runner import get_dist_info
+from mmengine.dist.utils import get_dist_info
 from mmcv.utils import Registry, build_from_cfg, digit_version
 from torch.utils.data import DataLoader
 

diff --git a/mmcls/datasets/cifar.py b/mmcls/datasets/cifar.py
@@ -5,7 +5,7 @@
 
 import numpy as np
 import torch.distributed as dist
-from mmengine.runner import get_dist_info
+from mmengine.dist.utils import get_dist_info
 
 from .base_dataset import BaseDataset
 from .builder import DATASETS

diff --git a/mmcls/datasets/mnist.py b/mmcls/datasets/mnist.py
@@ -6,7 +6,7 @@
 import numpy as np
 import torch
 import torch.distributed as dist
-from mmengine.runner import get_dist_info, master_only
+from mmengine.dist.utils import get_dist_info, master_only
 
 from .base_dataset import BaseDataset
 from .builder import DATASETS

diff --git a/mmcls/datasets/samplers/repeat_aug.py b/mmcls/datasets/samplers/repeat_aug.py
@@ -1,7 +1,7 @@
 import math
 
 import torch
-from mmengine.runner import get_dist_info
+from mmengine.dist.utils import get_dist_info
 from torch.utils.data import Sampler
 
 from mmcls.core.utils import sync_random_seed

diff --git a/tools/kfold-cross-valid.py b/tools/kfold-cross-valid.py
@@ -11,7 +11,7 @@
 import mmcv
 import torch
 from mmengine.config import Config, DictAction
-from mmengine.runner import get_dist_info, init_dist
+from mmengine.dist.utils import get_dist_info, init_dist
 
 from mmcls import __version__
 from mmcls.apis import init_random_seed, set_random_seed, train_model

diff --git a/tools/train.py b/tools/train.py
@@ -10,7 +10,7 @@
 import torch
 import torch.distributed as dist
 from mmengine.config import Config, DictAction
-from mmengine.runner import get_dist_info, init_dist
+from mmengine.dist.utils import get_dist_info, init_dist
 
 from mmcls import __version__
 from mmcls.apis import init_random_seed, set_random_seed, train_model