[Dependency] Update mmcv dependency version (#509)

* Update mmcv dependency version

* Add code info in some metafiles
pull/512/head
Ma Zerun 2021-11-02 18:08:30 +08:00 committed by GitHub
parent b8c3825bb3
commit d1473e4a7f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 18 additions and 9 deletions

View File

@ -15,6 +15,9 @@ Collections:
Title: 'Res2Net: A New Multi-scale Backbone Architecture'
URL: https://arxiv.org/pdf/1904.01169.pdf
README: configs/res2net/README.md
Code:
URL: https://github.com/open-mmlab/mmclassification/blob/v0.17.0/mmcls/models/backbones/res2net.py
Version: v0.17.0
Models:
- Name: res2net50-w14-s8_3rdparty_8xb32_in1k

View File

@ -12,6 +12,9 @@ Collections:
URL: https://arxiv.org/abs/2101.11986
Title: "Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet"
README: configs/t2t_vit/README.md
Code:
URL: https://github.com/open-mmlab/mmclassification/blob/v0.17.0/mmcls/models/backbones/t2t_vit.py
Version: v0.17.0
Models:
- Name: t2t-vit-t-14_3rdparty_8xb64_in1k

View File

@ -15,6 +15,9 @@ Collections:
URL: https://arxiv.org/pdf/2010.11929.pdf
Title: 'An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale'
README: configs/vision_transformer/README.md
Code:
URL: https://github.com/open-mmlab/mmclassification/blob/v0.17.0/mmcls/models/backbones/vision_transformer.py
Version: v0.17.0
Models:
- Name: vit-base-p16_in21k-pre-3rdparty_ft-64xb64_in1k-384

View File

@ -1,5 +1,5 @@
ARG PYTORCH="1.6.0"
ARG CUDA="10.1"
ARG PYTORCH="1.8.1"
ARG CUDA="10.2"
ARG CUDNN="7"
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
@ -13,7 +13,7 @@ RUN apt-get update && apt-get install -y ffmpeg libsm6 libxext6 git ninja-build
&& rm -rf /var/lib/apt/lists/*
# Install MMCV
RUN pip install mmcv-full==1.3.8 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
RUN pip install mmcv-full==1.3.16 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.1/index.html
# Install MMClassification
RUN conda clean --all

View File

@ -1,9 +1,9 @@
ARG PYTORCH="1.6.0"
ARG CUDA="10.1"
ARG PYTORCH="1.8.1"
ARG CUDA="10.2"
ARG CUDNN="7"
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
ARG MMCV="1.3.8"
ARG MMCV="1.3.16"
ARG MMCLS="0.17.0"
ENV PYTHONUNBUFFERED TRUE

View File

@ -47,7 +47,7 @@ def digit_version(version_str: str, length: int = 4):
return tuple(release)
mmcv_minimum_version = '1.3.8'
mmcv_minimum_version = '1.3.16'
mmcv_maximum_version = '1.5.0'
mmcv_version = digit_version(mmcv.__version__)

View File

@ -1 +1 @@
mmcv-full>=1.3.8,<=1.5.0
mmcv-full>=1.3.16,<=1.5.0

View File

@ -1,3 +1,3 @@
mmcv>=1.3.8
mmcv>=1.3.16
torch
torchvision