@@ -464,7 +464,7 @@ def from_keras3(model, input_signature=None, opset=None, custom_ops=None, custom
464
464
465
465
466
466
valid_names = []
467
- for out in model .output_names :
467
+ for out in [ t . name for t in model .outputs ] :
468
468
if out in reverse_lookup :
469
469
valid_names .append (reverse_lookup [out ])
470
470
else :
@@ -481,9 +481,6 @@ def from_keras3(model, input_signature=None, opset=None, custom_ops=None, custom
481
481
with tf .device ("/cpu:0" ):
482
482
frozen_graph , initialized_tables = \
483
483
tf_loader .from_trackable (model , concrete_func , input_names , output_names , large_model )
484
-
485
- for node in frozen_graph .node :
486
- print (node .name , node .op )
487
484
model_proto , external_tensor_storage = _convert_common (
488
485
frozen_graph ,
489
486
name = model .name ,
@@ -504,9 +501,6 @@ def from_keras3(model, input_signature=None, opset=None, custom_ops=None, custom
504
501
tensors_to_rename = tensors_to_rename ,
505
502
initialized_tables = initialized_tables ,
506
503
output_path = output_path )
507
-
508
- #print(model_proto)
509
-
510
504
return model_proto , external_tensor_storage
511
505
512
506
def from_keras (model , input_signature = None , opset = None , custom_ops = None , custom_op_handlers = None ,
0 commit comments