Skip to content

Pretrained weights

Pre-release
Pre-release
Compare
Choose a tag to compare
@alvinwan alvinwan released this 23 Mar 09:08
· 291 commits to master since this release
0.0.1
c8926e0

The following checkpoints were produced by training with this repository.

Includes checkpoints for the following neural networks:

  • WideResNet28x10 on TinyImagenet200
  • ResNet18 on CIFAR10, CIFAR100, TinyImagenet200
  • ResNet10 on CIFAR10, CIFAR100

Includes checkpoints for the following NBDTs, trained with induced hierarchies:

  • WideResNet28x10 on CIFAR10, CIFAR100, TinyImagenet200
  • ResNet18 on CIFAR10, CIFAR100, TinyImagenet200

Includes checkpoints for the following NBDTs, trained with the wordnet hierarchy (not currently load-able using --pretrained. Use --path-resume instead, after downloading manually):

  • WideResNet28x10 on CIFAR10

Note that we initialized with pytorchcv's pretrained WideResNet28x10 when training NBDTs on CIFAR10 and CIFAR100.

Assets 14
Loading