fix
parent
c71ed86d0e
commit
efb5851add
mmcls
apis
datasets
|
@ -10,7 +10,7 @@ import numpy as np
|
|||
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,
|
||||
|
|
|
@ -7,7 +7,7 @@ from functools import partial
|
|||
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
|
||||
|
||||
|
|
|
@ -5,7 +5,7 @@ import pickle
|
|||
|
||||
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
|
||||
|
|
|
@ -6,7 +6,7 @@ import os.path as osp
|
|||
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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -11,7 +11,7 @@ from pathlib import Path
|
|||
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
|
||||
|
|
|
@ -10,7 +10,7 @@ import mmcv
|
|||
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
|
||||
|
|
Loading…
Reference in New Issue