You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Introduce a command-line argument --flops, supporting model FLOPs computation and display.
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.)
Enhance support for loading pre-trained weights on timm models. Loading Priority: --pretrain > model_local_paths > timm.create_model(..., pretrained=True).
Introduce a command-line argument --clear_output_dir/-co, with which the specified --output_dir will be cleared first.
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]).
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).