mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcls/apis/test.py
View File

@ -10,7 +10,7 @@ import numpy as np
import torch import torch
import torch.distributed as dist import torch.distributed as dist
from mmcv.image import tensor2imgs 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, def single_gpu_test(model,

2
mmcls/datasets/builder.py
View File

@ -7,7 +7,7 @@ from functools import partial
import numpy as np import numpy as np
import torch import torch
from mmcv.parallel import collate 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 mmcv.utils import Registry, build_from_cfg, digit_version
from torch.utils.data import DataLoader from torch.utils.data import DataLoader

2
mmcls/datasets/cifar.py
View File

@ -5,7 +5,7 @@ import pickle
import numpy as np import numpy as np
import torch.distributed as dist 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 .base_dataset import BaseDataset
from .builder import DATASETS from .builder import DATASETS

2
mmcls/datasets/mnist.py
View File

@ -6,7 +6,7 @@ import os.path as osp
import numpy as np import numpy as np
import torch import torch
import torch.distributed as dist 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 .base_dataset import BaseDataset
from .builder import DATASETS from .builder import DATASETS

2
mmcls/datasets/samplers/repeat_aug.py
View File

@ -1,7 +1,7 @@
import math import math
import torch import torch
from mmengine.runner import get_dist_info from mmengine.dist.utils import get_dist_info
from torch.utils.data import Sampler from torch.utils.data import Sampler
from mmcls.core.utils import sync_random_seed from mmcls.core.utils import sync_random_seed

2
tools/kfold-cross-valid.py
View File

@ -11,7 +11,7 @@ from pathlib import Path
import mmcv import mmcv
import torch import torch
from mmengine.config import Config, DictAction 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 import __version__
from mmcls.apis import init_random_seed, set_random_seed, train_model from mmcls.apis import init_random_seed, set_random_seed, train_model

2
tools/train.py
View File

@ -10,7 +10,7 @@ import mmcv
import torch import torch
import torch.distributed as dist import torch.distributed as dist
from mmengine.config import Config, DictAction 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 import __version__
from mmcls.apis import init_random_seed, set_random_seed, train_model from mmcls.apis import init_random_seed, set_random_seed, train_model