pull/1952/head
wemoveon2 2024-10-01 11:23:41 -04:00
parent c71ed86d0e
commit efb5851add
No known key found for this signature in database
GPG Key ID: BEE790E675277B07
7 changed files with 7 additions and 7 deletions

View File

@ -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,

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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