ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 02:32:11 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix openmmlab cifar10 example raises ModuleNotFoundError (update dependency)

Browse Source
This commit is contained in:
allegroai 2023-05-11 15:55:34 +03:00
parent 1927e6578c
commit a3b4e1ec82
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
examples/frameworks/openmmlab/openmmlab_cifar10.py
View File

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

2
examples/frameworks/openmmlab/requirements.txt
View File

@ -1,4 +1,4 @@
clearml
mmcv>=1.5.1
mmcv>=1.5.1,<2.0.0
torch
torchvision