Deconvolution layer problem

Hello.

I have the follow problem:

I have the converted from Caffe(with deconvolutional layer) model which work correct if I run it with python and coremltools. But if I try to load it on Iphone the loader return assertation error:

/BuildRoot/Library/Caches/com.apple.xbs/Sources/MetalImage/MetalImage-82.1/MPSNeuralNetwork/Filters/MPSCNNConvolutionTranspose.mm:207: failed assertion `number of groups must be > 0'

I now that nGroups correct (=2)

Caffe section:

layer {
  name: "upscore"
  type: "Deconvolution"
  bottom: "conv10_new"
  top: "upscore"
  param {
  lr_mult: 0
  decay_mult: 0
  }
  convolution_param {
  num_output: 2
  bias_term: false
  pad: 16
  kernel_size: 64
  group: 2
  stride: 32
  weight_filler {
  type: "bilinear"
  }
  }
}

Mlmodel spec section:

  layers {
    name: "upscore"
    input: "conv10_new"
    output: "upscore"
    convolution {
      outputChannels: 2
      kernelChannels: 1
      nGroups: 2
      kernelSize: 64
      kernelSize: 64
      stride: 32
      stride: 32
      dilationFactor: 1
      dilationFactor: 1
      valid {
        paddingAmounts {
          borderAmounts {
            startEdgeSize: 16
            endEdgeSize: 16
          }
          borderAmounts {
            startEdgeSize: 16
            endEdgeSize: 16
          }
        }
      }
      isDeconvolution: true
      weights {

Any ideas?

Thanks!

PS. coremltools convert Caffe deconvolution layer not totally correct, it needed to swap outputChannels and kernelChannels (reshape weights in caffe model). For example in this sample caffe model has weights shape (2, 1, 64, 64) but coremltools expect (1, 2, 64, 64)