Skip to content

v0.1.1: More in1k models.

Compare
Choose a tag to compare
@dbolya dbolya released this 13 Jun 01:12
· 5 commits to main since this release

[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.