v0.1.1: More in1k models.
[2023.06.12] v0.1.1
- Added the ability to specify multiple pretrained checkpoints per architecture (specify with
checkpoint=<ckpt_name>
). - Added the ability to pass
strict=False
to a pretrained model so that you can use a different number of classes. Note: when changing the number of classes, the head layer will be reset. - Released all in1k finetuned models.