def get_network_def_from_model(self, model):
'''
please refer to get_network_def_from_model(...) in functions.py
'''
return fns.get_network_def_from_model(model, self.input_data_shape)
help='model architecture: ' + ' | '.join(model_names) +
' (default: alexnet)')
if __name__ == '__main__':
args = arg_parser.parse_args()
print(args)
build_lookup_table = True
lookup_table_path = args.dir
model_arch = args.arch
print('Load', model_arch)
print('--------------------------------------')
model = models.__dict__[model_arch](num_classes=NUM_CLASSES)
network_def = fns.get_network_def_from_model(model, INPUT_DATA_SHAPE)
for layer_name, layer_properties in network_def.items():
print(layer_name)
print(' ', layer_properties, '\n')
print('-------------------------------------------')
num_w = fns.compute_resource(network_def, 'WEIGHTS')
flops = fns.compute_resource(network_def, 'FLOPS')
num_param = fns.compute_resource(network_def, 'WEIGHTS')
print('Number of FLOPs: ', flops)
print('Number of weights: ', num_w)
print('Number of parameters: ', num_param)
print('-------------------------------------------')
model = model.cuda()
