I’ve added x.expand() in PyTorch and I do see the expansion layer in the ONNX model:
However, after parsing, the resulting .har file has this layer removed:
Resulting in a similar error:
```
Traceback (most recent call last):
File “/home/yuri/Source/hailo/hailo/bin/hailo”, line 7, in
sys.exit(main())
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_client/tools/cmd_utils/main.py”, line 111, in main
ret_val = client_command_runner.run()
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_client/tools/cmd_utils/base_utils.py”, line 68, in run
return self._run(argv)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_client/tools/cmd_utils/base_utils.py”, line 89, in _run
return args.func(args)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_client/tools/optimize_cli.py”, line 120, in run
self._runner.optimize(dataset, work_dir=args.work_dir)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_common/states/states.py”, line 16, in wrapped_func
return func(self, *args, **kwargs)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_client/runner/client_runner.py”, line 2206, in optimize
result = self._optimize(
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_common/states/states.py”, line 16, in wrapped_func
return func(self, *args, **kwargs)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_client/runner/client_runner.py”, line 2025, in _optimize
checkpoint_info = self._sdk_backend.full_quantization(
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_client/sdk_backend/sdk_backend.py”, line 1138, in full_quantization
new_checkpoint_info = self._full_acceleras_run(
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_client/sdk_backend/sdk_backend.py”, line 1376, in _full_acceleras_run
new_checkpoint_info = self._optimization_flow_runner(optimization_flow, checkpoint_info)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_sdk_client/sdk_backend/sdk_backend.py”, line 2029, in _optimization_flow_runner
optimization_flow.run()
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/tools/orchestator.py”, line 239, in wrapper
return func(self, *args, **kwargs)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/flows/optimization_flow.py”, line 357, in run
step_func()
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/tools/subprocess_wrapper.py”, line 153, in parent_wrapper
func(self, *args, **kwargs)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/flows/optimization_flow.py”, line 374, in step1
self._update_fp_data()
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/flows/optimization_flow.py”, line 360, in _update_fp_data
self.set_fp_params(self._model.export_weights())
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/acceleras/model/hailo_model/hailo_model.py”, line 604, in export_weights
return self._export_npz(mode=NpzExportMode.WEIGHTS, include_shared_weights=include_shared_weights)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/acceleras/model/hailo_model/hailo_model.py”, line 613, in _export_npz
self.build(shapes)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/keras/src/layers/layer.py”, line 226, in build_wrapper
original_build_method(*args, **kwargs)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/acceleras/utils/distributed_utils.py”, line 122, in wrapper
res = func(self, *args, **kwargs)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/acceleras/model/hailo_model/hailo_model.py”, line 1260, in build
layer.build(layer.input_shapes)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/keras/src/layers/layer.py”, line 226, in build_wrapper
original_build_method(*args, **kwargs)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/acceleras/hailo_layers/base_hailo_layer.py”, line 1570, in build
self.verify_layer_inputs_shape(input_shape)
File “/home/yuri/Source/hailo/hailo/lib/python3.10/site-packages/hailo_model_optimization/acceleras/hailo_layers/base_hailo_conv_add.py”, line 204, in verify_layer_inputs_shape
raise InvalidInputShape(
hailo_model_optimization.acceleras.utils.acceleras_exceptions.InvalidInputShape: Input shapes [[None, 120, 208, 16], [None, 120, 208, 1]] doesn’t match each other in octonet/conv2
```