Model training GPU

Please have a look at the following post with the same issue and an answer from my colleague that solved it.

Hailo Community - Problem with model optimization

Please let us know whether this worked for you as well.