Skip to content

Releases: alvinwan/neural-backed-decision-trees

Improvements to Vis, Hierarchy, Inference Scripts

31 Mar 04:53
@alvinwan alvinwan
0.0.4
189b191
Compare
Choose a tag to compare
Improvements to Vis, Hierarchy, Inference Scripts Pre-release
Pre-release 
0.0.4

version bump
Assets 2
Loading

Initial Release

25 Mar 23:53
@alvinwan alvinwan
0.0.2
4dc43d8
Compare
Choose a tag to compare
Initial Release Pre-release
Pre-release
  • Code release on PyPi for testing
  • Includes 3 console scripts: nbdt, nbdt-hierarchy, nbdt-wnids
  • A number of default hierarchies and WordNet IDs for CIFAR10, CIFAR100, TinyImagenet, Imagenet
Assets 2
Loading

Pretrained weights

23 Mar 09:08
@alvinwan alvinwan
0.0.1
c8926e0
Compare
Choose a tag to compare
Pretrained weights Pre-release
Pre-release

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