Skip to content

Take into account CHW and HWC data format

Olivier Antoni requested to merge oantoni/aidge_core:update_ai4hp into dev

Context

Model export to SMT32 fails because HWC data format is currently not supported:

  File "aidge_core/export_utils/node_export.py", line 38, in get_chan
    raise RuntimeError(f"Unknown dataformat: {dformat}")
RuntimeError: Unknown dataformat: hwc

Detailed major modifications

  • Add support for CHW and HWC data formats,
  • Add "assert" functions to avoid returning erroneous N/C/H/W values ​​if the number of Tensor dimensions is not equal to the length of the data format.

Merge request reports

Loading