Skip to content

fix : Update the compare_layers script

Context

Added small fixes and update to the comparison script:

  • Fixed Scheduler call
  • Removed unsupported typehints (type hint "set" and "dict" are only available for python > 3.9)
  • Added backend argument to choose between "cpu" and "cuda"

Modified files

  • compare_layers_aidge_onnx.py;

Merge request reports