hierarchy download=True

Author: alvinwan

nbdt/hierarchies/CIFAR10/graph-induced-ResNet10.json

@@ -1 +1 @@
-{"directed": true, "multigraph": false, "graph": {}, "nodes": [{"label": "airplane", "id": "n02691156"}, {"label": "car", "id": "n02958343"}, {"label": "bird", "id": "n01503061"}, {"label": "cat", "id": "n02121620"}, {"label": "deer", "id": "n02430045"}, {"label": "dog", "id": "n02084071"}, {"label": "frog", "id": "n01639765"}, {"label": "horse", "id": "n02374451"}, {"label": "ship", "id": "n04194289"}, {"label": "truck", "id": "n04490091"}, {"id": "f00000010"}, {"id": "f00000011"}, {"id": "f00000012"}, {"id": "f00000013"}, {"id": "f00000014"}, {"id": "f00000015"}, {"id": "f00000016"}, {"id": "f00000017"}, {"id": "f00000018"}], "links": [{"source": "f00000010", "target": "n02958343"}, {"source": "f00000010", "target": "n04490091"}, {"source": "f00000011", "target": "n04194289"}, {"source": "f00000011", "target": "f00000010"}, {"source": "f00000012", "target": "n02084071"}, {"source": "f00000012", "target": "n02374451"}, {"source": "f00000013", "target": "n02121620"}, {"source": "f00000013", "target": "n01639765"}, {"source": "f00000014", "target": "n02691156"}, {"source": "f00000014", "target": "f00000011"}, {"source": "f00000015", "target": "n02430045"}, {"source": "f00000015", "target": "f00000012"}, {"source": "f00000016", "target": "f00000013"}, {"source": "f00000016", "target": "f00000014"}, {"source": "f00000017", "target": "n01503061"}, {"source": "f00000017", "target": "f00000015"}, {"source": "f00000018", "target": "f00000016"}, {"source": "f00000018", "target": "f00000017"}]}
+{"directed": true, "multigraph": false, "graph": {}, "nodes": [{"label": "airplane", "id": "n02691156"}, {"label": "car", "id": "n02958343"}, {"label": "bird", "id": "n01503061"}, {"label": "cat", "id": "n02121620"}, {"label": "deer", "id": "n02430045"}, {"label": "dog", "id": "n02084071"}, {"label": "frog", "id": "n01639765"}, {"label": "horse", "id": "n02374451"}, {"label": "ship", "id": "n04194289"}, {"label": "truck", "id": "n04490091"}, {"label": "motor_vehicle", "id": "n03791235"}, {"label": "vehicle", "id": "n04524313"}, {"label": "placental", "id": "n01886756"}, {"label": "vertebrate", "id": "n01471682"}, {"label": "conveyance", "id": "n03100490"}, {"label": "mammal", "id": "n01861778"}, {"label": "whole", "id": "n00003553"}, {"label": "chordate", "id": "n01466257"}, {"label": "object", "id": "n00002684"}], "links": [{"source": "n03791235", "target": "n02958343"}, {"source": "n03791235", "target": "n04490091"}, {"source": "n04524313", "target": "n04194289"}, {"source": "n04524313", "target": "n03791235"}, {"source": "n01886756", "target": "n02084071"}, {"source": "n01886756", "target": "n02374451"}, {"source": "n01471682", "target": "n02121620"}, {"source": "n01471682", "target": "n01639765"}, {"source": "n03100490", "target": "n02691156"}, {"source": "n03100490", "target": "n04524313"}, {"source": "n01861778", "target": "n02430045"}, {"source": "n01861778", "target": "n01886756"}, {"source": "n00003553", "target": "n01471682"}, {"source": "n00003553", "target": "n03100490"}, {"source": "n01466257", "target": "n01503061"}, {"source": "n01466257", "target": "n01861778"}, {"source": "n00002684", "target": "n00003553"}, {"source": "n00002684", "target": "n01466257"}]}

nbdt/hierarchy.py

@@ -349,7 +349,7 @@ def generate_hierarchy_vis(args):
     dataset = None
     if args.dataset:
         cls = getattr(data, args.dataset)
-        dataset = cls(root='./data', train=False, download=False)
+        dataset = cls(root='./data', train=False, download=True)
 
     color_info = get_color_info(
         G,