Skip to content

v0.8.0

Compare
Choose a tag to compare
@horrible-dong horrible-dong released this 02 Dec 07:03
· 29 commits to main since this release

Dec 2, 2023

  1. Introduce a command-line argument --flops, supporting model FLOPs computation and display.
  2. Introduce a command-line argument --pretrain/-p. To temporarily use a pre-trained weight path, you can specify it by command-line argument --pretrain / -p. For long-term use of a pre-trained weight path, it is preferable to write it in _pretrain_.py. Loading Priority: --pretrain > model_local_paths > model_urls. (Please carefully distinguish between --pretrain/-p and --resume/-r.)
  3. Enhance support for loading pre-trained weights on timm models. Loading Priority: --pretrain > model_local_paths > timm.create_model(..., pretrained=True).
  4. Introduce a command-line argument --clear_output_dir/-co, with which the specified --output_dir will be cleared first.
  5. In qtcls/datasets/__init__.py, num_classes -> _num_classes. Add _image_size where users can optionally specify the dataset's default target image size. During data preprocessing, images will be automatically scaled to the target image size. You can also specify the target image size by --image_size (priority: --image_size > _image_size[dataset_name]).
  6. Fix the bug where the image size cannot adapt to changes when using fake data (--dummy) of different datasets such as CIFAR (32×32) and ImageNet (224×224).
  7. Redefine imagenet.py and add more guidelines.
  8. Rename plot_utils.py to log_plot.py.
  9. Minor bug fixes and improvements.
  10. Update docs.