I am unable to save the trace model to a coreml file. My code:
input_shape = ct.Shape(shape=[1, 3, 256, 256]) converted = ct.convert( traced_model, in