[Dependency] Update mmcv dependency version (#509)
* Update mmcv dependency version * Add code info in some metafilespull/512/head
parent
b8c3825bb3
commit
d1473e4a7f
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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__)
|
||||
|
||||
|
|
|
@ -1 +1 @@
|
|||
mmcv-full>=1.3.8,<=1.5.0
|
||||
mmcv-full>=1.3.16,<=1.5.0
|
||||
|
|
|
@ -1,3 +1,3 @@
|
|||
mmcv>=1.3.8
|
||||
mmcv>=1.3.16
|
||||
torch
|
||||
torchvision
|
||||
|
|
Loading…
Reference in New Issue