Releases: horrible-dong/QTClassification
Releases · horrible-dong/QTClassification
v0.3.0
Apr 22, 2023
- Highlight the usage of configs in README.md. We recommend writing arguments into a config file.
- Fix: Support automatic mixed precision (amp) during evaluation.
- Support log plotting. See qtcls/utils/plot_utils.py.
- Support using
args.image_size
to set the input image size. - Support using
args.model_kwargs
to set the model arguments in dict format. Here is an example in configs/_demo_.py.args.model_kwargs
will be passed in qtcls/models/__init__.py, line 57. - Minor adjustments.
v0.2.0
Apr 10, 2023
- Enrich the architectures of
Swin Transformer
andVision Transformer (timm)
. - Support new models:
Swin Transformer V2
,CaiT
,LeViT
,MLP-Mixer
andPoolFormer
. - Optimize cosine lr scheduler. Now, the learning rate can be changed not only epoch by epoch, but also step by step.
- Starting from this version (v0.2.0), if you want to use your config file, set
--config
/-c
to your config file path,
likeconfigs/_demo_.py
. Note that--config
supports any file system path. Please see configs for more details. - Give hints for resolving some exceptions/errors.
- Minor bug fixes and improvements.
v0.1.0
Apr 3, 2023
- Support almost all popular datasets and models.
- Support single-gpu and multi-gpu (DDP) training.
- Support automatic mixed precision training.
- Fully functional workflow.
- Highly customizable pipeline.