From fb0a82d5cddc35260a66851c41cbe0161d091ee1 Mon Sep 17 00:00:00 2001
From: horrible-dong <954513308@qq.com>
Date: Wed, 6 Mar 2024 16:52:55 +0800
Subject: [PATCH] [20240306] Version upgrade -> v0.9.0

---
 README.md         | 14 ++++++++------
 README_zh-CN.md   | 15 +++++++++------
 qtcls/__init__.py |  2 +-
 3 files changed, 18 insertions(+), 13 deletions(-)

diff --git a/README.md b/README.md
index 9092a6a..564480d 100644
--- a/README.md
+++ b/README.md
@@ -3,7 +3,7 @@ QTClassification
 
 **A lightweight and extensible toolbox for image classification**
 
-[![version](https://img.shields.io/badge/Version-0.9.0--dev-brightgreen)](https://github.com/horrible-dong/QTClassification)
+[![version](https://img.shields.io/badge/Version-0.9.0-brightgreen)](https://github.com/horrible-dong/QTClassification)
 &emsp;[![docs](https://img.shields.io/badge/Docs-Latest-orange)](https://github.com/horrible-dong/QTClassification/blob/main/README.md)
 &emsp;[![license](https://img.shields.io/badge/License-Apache--2.0-blue)](https://github.com/horrible-dong/QTClassification/blob/main/LICENSE)
 
@@ -85,9 +85,11 @@ python -m torch.distributed.launch --nproc_per_node=2 --use_env main.py \
   --output_dir ./runs/__tmp__
 ```
 
-The `cifar10` dataset and `resnet50` pretrained weights will be automatically downloaded. The `cifar10` dataset will be
-downloaded to `./data`. During the training, the config file, checkpoints, logs and other outputs will be stored in
-`./runs/__tmp__`.
+The `cifar10` dataset and `resnet50` pretrained weights will be automatically downloaded. Please keep the network
+accessible. The `cifar10` dataset will be downloaded to `./data`. The `resnet50` pretrained weights will be downloaded
+to `~/.cache/torch/hub/checkpoints`.
+
+During training, the config file, checkpoints, logs, and other outputs will be stored in `./runs/__tmp__.`
 
 **Evaluation**
 
@@ -297,7 +299,7 @@ Currently supported argument `--model` / `-m`:
 Set the argument `--model_lib` to `timm`.
 
 Currently supported argument `--model` / `-m`:  
-All supported. Please refer to `timm` for the specific model names.
+All supported. Please refer to `timm` for specific model names.
 
 ## License
 
@@ -323,4 +325,4 @@ If you find QTClassification Toolbox useful in your research, please consider ci
     howpublished={\url{https://github.com/horrible-dong/QTClassification}},
     year={2023}
 }
-```
+```
\ No newline at end of file
diff --git a/README_zh-CN.md b/README_zh-CN.md
index 8b7b1e4..b2381d8 100644
--- a/README_zh-CN.md
+++ b/README_zh-CN.md
@@ -3,7 +3,7 @@ QTClassification
 
 **轻量可扩展的图像分类工具箱**
 
-[![version](https://img.shields.io/badge/Version-0.9.0--dev-brightgreen)](https://github.com/horrible-dong/QTClassification)
+[![version](https://img.shields.io/badge/Version-0.9.0-brightgreen)](https://github.com/horrible-dong/QTClassification)
 &emsp;[![docs](https://img.shields.io/badge/Docs-Latest-orange)](https://github.com/horrible-dong/QTClassification/blob/main/README_zh-CN.md)
 &emsp;[![license](https://img.shields.io/badge/License-Apache--2.0-blue)](https://github.com/horrible-dong/QTClassification/blob/main/LICENSE)
 
@@ -36,8 +36,8 @@ conda activate qtcls
 pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
 ```
 
-或者你可以参考[PyTorch官网](https://pytorch.org/get-started/previous-versions/)来安装其他版本。请注意，如果使用 pytorch ≥
-1.13，则需要 python ≥ 3.8。
+或者你可以参考[PyTorch官网](https://pytorch.org/get-started/previous-versions/)来安装其他版本。
+请注意，如果使用 pytorch ≥ 1.13，则需要 python ≥ 3.8。
 
 4. 安装必要的依赖。
 
@@ -86,8 +86,11 @@ python -m torch.distributed.launch --nproc_per_node=2 --use_env main.py \
   --output_dir ./runs/__tmp__
 ```
 
-`cifar10` 数据集和 `resnet50` 预训练权重会自动下载。`cifar10` 数据集会被下载到 `./data` 目录下。在训练过程中，
-配置文件（config），checkpoint文件（包含模型权重、优化器权重等）、日志文件（log）和其他输出都会被存放在 `./runs/__tmp__` 目录下。
+`cifar10` 数据集和 `resnet50` 预训练权重会自动下载，请保持网络畅通。`cifar10` 数据集会被下载到 `./data` 目录下，`resnet50`
+预训练权重会被下载到 `~/.cache/torch/hub/checkpoints` 目录下。若不使用预训练模型，请添加命令行参数 `--no_pretrain`。
+
+在训练过程中，配置文件（config），checkpoint 文件（包含模型权重、优化器权重等）、日志文件（log）和其他输出都会被存放在
+`./runs/__tmp__` 目录下。
 
 **验证**
 
@@ -312,4 +315,4 @@ QTClassification 基于 Apache 2.0 开源许可证。具体请看[开源许可
     howpublished={\url{https://github.com/horrible-dong/QTClassification}},
     year={2023}
 }
-```
+```
\ No newline at end of file
diff --git a/qtcls/__init__.py b/qtcls/__init__.py
index b2b6335..369634e 100644
--- a/qtcls/__init__.py
+++ b/qtcls/__init__.py
@@ -3,7 +3,7 @@
 # ********************************************
 # Copyright (c) QIU Tian. All rights reserved.
 
-__version__ = "v0.9.0-dev"
+__version__ = "v0.9.0"
 __git_url__ = "https://github.com/horrible-dong/QTClassification"
 
 from .criterions import build_criterion