Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet
MIT License
Published by alvinwan over 4 years ago
Published by alvinwan over 4 years ago
nbdt
, nbdt-hierarchy
, nbdt-wnids
Published by alvinwan over 4 years ago
The following checkpoints were produced by training with this repository.
Includes checkpoints for the following neural networks:
Includes checkpoints for the following NBDTs, trained with induced hierarchies:
Includes checkpoints for the following NBDTs, trained with the wordnet hierarchy (not currently load-able using --pretrained
. Use --path-resume
instead, after downloading manually):
Note that we initialized with pytorchcv
's pretrained WideResNet28x10 when training NBDTs on CIFAR10 and CIFAR100.