Skip to content
Snippets Groups Projects

Correct logic for Node's parents/children association with inputs/outputs in aidge_core/show_graphview.py.

Merged Iryna de Albuquerque Silva requested to merge idealbuq/aidge_core:dev into dev
1 file
+ 2
2
Compare changes
  • Side-by-side
  • Inline
@@ -129,7 +129,7 @@ def _create_dict(ordered_nodes : list[aidge_core.Node], write_trainable_params_e
@@ -129,7 +129,7 @@ def _create_dict(ordered_nodes : list[aidge_core.Node], write_trainable_params_e
if params_file_format=='npz':
if params_file_format=='npz':
np.savez_compressed(Path(path_trainable_params, node.name()), **{node.name() : node.get_operator().get_output(0)})
np.savez_compressed(Path(path_trainable_params, node.name()), **{node.name() : node.get_operator().get_output(0)})
node_dict['tensor_data'] = Path(path_trainable_params, node.name() + '.npz')
node_dict['tensor_data'] = str(Path(path_trainable_params, node.name() + '.npz'))
elif params_file_format=='json':
elif params_file_format=='json':
tensor = np.array(node.get_operator().get_output(0))
tensor = np.array(node.get_operator().get_output(0))
@@ -145,7 +145,7 @@ def _create_dict(ordered_nodes : list[aidge_core.Node], write_trainable_params_e
@@ -145,7 +145,7 @@ def _create_dict(ordered_nodes : list[aidge_core.Node], write_trainable_params_e
with open(Path(path_trainable_params, node.name() + '.json'), 'w') as fp:
with open(Path(path_trainable_params, node.name() + '.json'), 'w') as fp:
json.dump(tensor_dict, fp, indent=4)
json.dump(tensor_dict, fp, indent=4)
node_dict['tensor_data'] = Path(path_trainable_params, node.name() + '.json')
node_dict['tensor_data'] = str(Path(path_trainable_params, node.name() + '.json'))
else:
else:
raise Exception("File format to write trainable parameters not recognized.")
raise Exception("File format to write trainable parameters not recognized.")
Loading