From 973a4fb9ca0dbf281f0ef34470785a35fef7eba9 Mon Sep 17 00:00:00 2001 From: Marek Drozdowski Date: Mon, 14 Jan 2019 13:31:31 -0800 Subject: [PATCH] print model url --- tftrt/examples/image-classification/image_classification.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tftrt/examples/image-classification/image_classification.py b/tftrt/examples/image-classification/image_classification.py index a02a0f059..b821da41d 100644 --- a/tftrt/examples/image-classification/image_classification.py +++ b/tftrt/examples/image-classification/image_classification.py @@ -175,6 +175,10 @@ def get_input_dims(self): def get_num_classes(self): return self.num_classes + def get_url(self): + return self.url + + def get_netdef(model): """ Creates the dictionary NETS with model names as keys and NetDef as values. @@ -606,6 +610,7 @@ def print_dict(input_dict, str='', scale=None): print('{}{}'.format(headline, '%.1f'%v if type(v)==float else v)) print_dict(vars(args)) + print("url: " + get_netdef(args.model).get_url()) print_dict(num_nodes, str='num_nodes') print_dict(graph_sizes, str='graph_size(MB)', scale=1./(1<<20)) print_dict(times, str='time(s)')