Fix for onnx optimization on lightweight open pose learner

Illia Oleksiienkorequested to merge
fix-onnx-conv into master

Created by: tsampazk

  • Added condition on __convert_to_onnx method to account for device

Merge request reports