From a3b4e1ec82ac1be82d1ac988bfad286c681036d5 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Thu, 11 May 2023 15:55:34 +0300
Subject: [PATCH] Fix openmmlab cifar10 example raises ModuleNotFoundError
 (update dependency)

---
 examples/frameworks/openmmlab/openmmlab_cifar10.py | 5 ++---
 examples/frameworks/openmmlab/requirements.txt     | 2 +-
 2 files changed, 3 insertions(+), 4 deletions(-)

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