diff --git a/examples/frameworks/openmmlab/openmmlab_cifar10.py b/examples/frameworks/openmmlab/openmmlab_cifar10.py index 4255dd07..bef3105d 100644 --- a/examples/frameworks/openmmlab/openmmlab_cifar10.py +++ b/examples/frameworks/openmmlab/openmmlab_cifar10.py @@ -3,12 +3,11 @@ import torch.nn as nn import torch.nn.functional as F import torch.optim as optim import torchvision.transforms as transforms -from torch.utils.data import DataLoader -from torchvision.datasets import CIFAR10 - from mmcv.parallel import MMDataParallel from mmcv.runner import EpochBasedRunner from mmcv.utils import get_logger +from torch.utils.data import DataLoader +from torchvision.datasets import CIFAR10 class Model(nn.Module): diff --git a/examples/frameworks/openmmlab/requirements.txt b/examples/frameworks/openmmlab/requirements.txt index 010df3e4..fae81627 100644 --- a/examples/frameworks/openmmlab/requirements.txt +++ b/examples/frameworks/openmmlab/requirements.txt @@ -1,4 +1,4 @@ clearml -mmcv>=1.5.1 +mmcv>=1.5.1,<2.0.0 torch torchvision