https://github.com/alvinwan/neural-backed-decision-trees

sort by:
Revision Author Date Message Commit Date
9cd5de9 removed unused models 23 March 2020, 04:48:22 UTC
1d58119 added support for --hierarchy flag instead of --path-graph 23 March 2020, 04:43:25 UTC
9e50d26 🌞 23 March 2020, 04:35:19 UTC
126a221 added bash scripts for modification, both for networks with pretrained and those without pretrained ccheckpoints 23 March 2020, 04:32:05 UTC
6208f3e way fewer scripts + fleshed out dev section 23 March 2020, 04:12:31 UTC
762db54 simpler training/eval 23 March 2020, 03:36:35 UTC
c06de4b nbdt-hierarchy now supports model name 23 March 2020, 00:52:42 UTC
4a93c4b always log graph hierarchy 23 March 2020, 00:06:37 UTC
c34c659 added nbdt eval script 22 March 2020, 15:06:05 UTC
729e2dc made generate_hiehrarchy and generate_wnids globally accessible scripts 22 March 2020, 14:43:39 UTC
c8b20bb combined generate_vis and gegnerate_hierarchies 22 March 2020, 14:17:22 UTC
42660b5 combined test + generate hierarchies 22 March 2020, 14:06:31 UTC
cf8efe6 oops, 'net' 22 March 2020, 13:37:21 UTC
1ff514a support pretrained kwarg for nbdt models 22 March 2020, 13:37:21 UTC
ca79168 added support for pretrained models 22 March 2020, 13:37:21 UTC
79c5d2d list all imports 22 March 2020, 11:53:19 UTC
a1b3841 use wrn for example isntead 22 March 2020, 11:45:48 UTC
23120df re-added cifar10/cifar100/tinyimagennet friendly resnets 22 March 2020, 11:42:23 UTC
c2c695e added test for nbdt model -> nbdt loss check (fixed analysis) 22 March 2020, 11:29:36 UTC
fc3b206 update with new interface 22 March 2020, 10:53:13 UTC
0ff05c4 updated scripts to use torchvision model name 22 March 2020, 10:48:02 UTC
2df40ac Merge branch 'master' of github.com:alvinwan/pytorch-cifar-plus into public 22 March 2020, 10:36:27 UTC
2872d56 make scripts clearer 22 March 2020, 10:25:46 UTC
83795eb removed static method initializations 22 March 2020, 10:25:40 UTC
c31005b new interface, no staticmethod 22 March 2020, 10:23:40 UTC
9249af7 removed all models in torchvision + using torchvision version 22 March 2020, 10:19:38 UTC
84141bf flesh out quickstart 22 March 2020, 03:39:04 UTC
5452402 Update README.md 21 March 2020, 13:08:24 UTC
7ef839c Update README.md 21 March 2020, 12:44:39 UTC
eb8fae8 support hierarchy kwargg 21 March 2020, 12:23:30 UTC
2655133 filled in info for nbdt import/usage 21 March 2020, 12:12:36 UTC
c580c41 added python package setup + fixed nbdt interface 21 March 2020, 11:47:13 UTC
58ce5ff remove predicted from update_batch in analysis, making it easier to work with 20 March 2020, 11:02:17 UTC
e30ae5b added nbdt model wrappers + inference-only calls 20 March 2020, 10:52:41 UTC
63c0bb9 Added new loss/analytsis initializers 20 March 2020, 10:26:29 UTC
dfb4623 cleaned up default path graph, path wnids 20 March 2020, 10:13:16 UTC
b2328ca simpler gitignonre 20 March 2020, 10:00:03 UTC
d291c18 fix writing utils to new paths 20 March 2020, 09:56:45 UTC
c4608a7 moved templates from vis to nbdt/templates 20 March 2020, 09:48:37 UTC
10c972f move all hierarchies and wnidsd to nbdt/ 20 March 2020, 09:45:02 UTC
9f9e655 added cifar100 wnids 20 March 2020, 09:36:46 UTC
a9232db removed old tree 20 March 2020, 09:36:12 UTC
badb994 changed all hierarchy usage to hierarchies dir 20 March 2020, 09:35:44 UTC
7d4578e moved all hierarchies to hierarchies dir 20 March 2020, 09:31:16 UTC
8c57a94 fixed batch size 20 March 2020, 08:05:17 UTC
bd6ecc8 added tinyimagenet wrn 20 March 2020, 03:16:41 UTC
4aa2d49 break in get_centers 20 March 2020, 02:52:04 UTC
c12538e moar edits 20 March 2020, 02:46:22 UTC
28c8d69 declare checkpoint path to save at, fix readme/bash scripts 20 March 2020, 02:42:29 UTC
2dda220 *further total speedup by 20%, for loop sped up by 5x 20 March 2020, 01:28:30 UTC
f2e6318 25% faster soft loss, vectorized first loop 20 March 2020, 01:28:25 UTC
210a88d decay every 15 March 2020, 05:48:39 UTC
b863ff8 support variable input size for cifar too 14 March 2020, 02:44:03 UTC
4cf09f0 support pytorch-image-models 13 March 2020, 08:32:49 UTC
e412033 fix printed names for inference 13 March 2020, 06:55:53 UTC
dfa64b2 do not add default paths if reguglar trainningg 12 March 2020, 22:58:12 UTC
ffafecb fill in partial 12 March 2020, 11:14:40 UTC
3d1f2a1 fixed input size arg 12 March 2020, 10:51:04 UTC
1acfbd4 remove .clone() call 12 March 2020, 10:19:24 UTC
aad56e2 hypothetical integration steps 12 March 2020, 09:55:53 UTC
a419d75 added resnet versions of scripts 12 March 2020, 09:24:18 UTC
609e633 updated README with new flags 12 March 2020, 09:11:20 UTC
fefcede fix new names 12 March 2020, 09:11:13 UTC
252b1be rename analysis for nbdt + reduce redudnancy 12 March 2020, 09:05:36 UTC
45ddba6 support custom path for resuming 12 March 2020, 08:50:32 UTC
573b42c coerce with original so that changes are minimal 12 March 2020, 08:32:44 UTC
82223b4 specify custom loss, data, analysis keys from respective files 12 March 2020, 08:13:07 UTC
b6443b7 renamed utils -> nbdt 12 March 2020, 07:43:26 UTC
140a310 custom args for analysis 12 March 2020, 01:47:48 UTC
3dc0249 moved loss args to loss file 12 March 2020, 01:32:17 UTC
279aefa removed lr scheduler pow because it was useuless 12 March 2020, 01:15:50 UTC
8784572 removed old sanity checks (in favor of new ones) 12 March 2020, 01:14:47 UTC
0eec967 removed , , since non longer used 12 March 2020, 01:11:22 UTC
341c2e1 removed backbone from scripts 12 March 2020, 01:07:23 UTC
0ee6d65 remove unused custom datasets 12 March 2020, 01:01:36 UTC
3f52a96 instantiate custom criterionn dircetly, instead of callign model 12 March 2020, 00:58:52 UTC
d6766d5 renamed sup losses 12 March 2020, 00:28:48 UTC
30b1c2f removed jointNodes, jointTree, decisionTree 12 March 2020, 00:28:45 UTC
65b7d4c removed tree model 12 March 2020, 00:28:41 UTC
38c1259 moved custom losses to utils/loss 12 March 2020, 00:28:37 UTC
2554c88 rm 11 March 2020, 07:17:10 UTC
baa890f added new scripts and removed old ones 11 March 2020, 07:06:55 UTC
8d2fd03 Update README.md 11 March 2020, 07:05:36 UTC
c57e142 renamed to hierarchy 11 March 2020, 05:19:42 UTC
eee2e90 removed unused tree 11 March 2020, 05:19:27 UTC
7c4ea26 usin cifar UGH 11 March 2020, 04:29:38 UTC
b351fc9 support arbirary input sizes for wrn_cifar (but maybe gets bad acc) 11 March 2020, 04:28:39 UTC
b40b7b5 fixed default tinyiimiagenet input size 11 March 2020, 03:16:00 UTC
615c59a added dataset-agnonstic version of wrn cifar 05 March 2020, 09:45:06 UTC
600d4d9 fixed assertion for finetuning efficientnet 03 March 2020, 23:29:51 UTC
58e3d3d fixed efficientnet backbone load 02 March 2020, 07:06:24 UTC
a4b11d4 fixed effnet (expects first pos arg to be input size 02 March 2020, 06:01:43 UTC
a0dfaa5 remove extraneous dollar sign 02 March 2020, 05:10:25 UTC
bd63820 added lr schedule pow 02 March 2020, 04:44:23 UTC
95c525c default input size 02 March 2020, 03:14:42 UTC
84b644f crap int 02 March 2020, 03:08:54 UTC
efbddfd add input size to arg 02 March 2020, 03:08:04 UTC
4fbb38f support efficientnet 02 March 2020, 01:49:47 UTC
948f817 added efficientnet 02 March 2020, 01:08:56 UTC
8739867 recognize linear weights in imgclsmob 02 March 2020, 00:09:51 UTC
back to top