Skip to content

Commit

Permalink
[20240306] Version upgrade -> v0.9.0
Browse files Browse the repository at this point in the history
  • Loading branch information
horrible-dong committed Mar 6, 2024
1 parent d7cd703 commit fb0a82d
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 13 deletions.
14 changes: 8 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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)
 [![docs](https://img.shields.io/badge/Docs-Latest-orange)](https://github.com/horrible-dong/QTClassification/blob/main/README.md)
 [![license](https://img.shields.io/badge/License-Apache--2.0-blue)](https://github.com/horrible-dong/QTClassification/blob/main/LICENSE)

Expand Down Expand Up @@ -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**

Expand Down Expand Up @@ -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

Expand All @@ -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}
}
```
```
15 changes: 9 additions & 6 deletions README_zh-CN.md
Original file line number Diff line number Diff line change
Expand Up @@ -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)
 [![docs](https://img.shields.io/badge/Docs-Latest-orange)](https://github.com/horrible-dong/QTClassification/blob/main/README_zh-CN.md)
 [![license](https://img.shields.io/badge/License-Apache--2.0-blue)](https://github.com/horrible-dong/QTClassification/blob/main/LICENSE)

Expand Down Expand Up @@ -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. 安装必要的依赖。

Expand Down Expand Up @@ -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__` 目录下。

**验证**

Expand Down Expand Up @@ -312,4 +315,4 @@ QTClassification 基于 Apache 2.0 开源许可证。具体请看[开源许可
howpublished={\url{https://github.com/horrible-dong/QTClassification}},
year={2023}
}
```
```
2 changes: 1 addition & 1 deletion qtcls/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit fb0a82d

Please sign in to comment.