onnx-coreml

Errors converting PyTorch Unet (“tiramisu”) into coreml, via onnx

半世苍凉 提交于 2019-12-23 04:06:42
问题 I'm trying to convert a pytorch "tiramisu" UNet (from: https://github.com/bfortuner/pytorch_tiramisu) to coreml, via onnx, and I'm getting this error in onnx-coreml's _operators.py : TypeError: Error while converting op of type: Concat. Error message: Unsupported axis 1 in input of shape Any thoughts about how I might work around this? The layers file is here, for reference: https://github.com/bfortuner/pytorch_tiramisu/blob/master/models/layers.py UPDATE 1: So, digging down further into this