ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 10:42:00 +00:00
Code Issues Actions 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.nn.functional as F
import torch.optim as optim import torch.optim as optim
import torchvision.transforms as transforms import torchvision.transforms as transforms
from torch.utils.data import DataLoader
from torchvision.datasets import CIFAR10
from mmcv.parallel import MMDataParallel from mmcv.parallel import MMDataParallel
from mmcv.runner import EpochBasedRunner from mmcv.runner import EpochBasedRunner
from mmcv.utils import get_logger from mmcv.utils import get_logger
from torch.utils.data import DataLoader
from torchvision.datasets import CIFAR10
class Model(nn.Module): class Model(nn.Module):

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

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