You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
import coremltools as ct
from coremltools.converters.mil.mil import types
coreml_model = ct.convert(
traced_model,
convert_to="mlprogram",
inputs=[ct.TensorType(shape=(1, 3, 16, 224, 224),dtype=types.float)]
)
coreml_model.save('yowo_tiny.mlmodel')
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\builder.py", line 168, in _add_op
new_op = op_cls(**kwargs)
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\operation.py", line 190, in init
self._validate_and_set_inputs(input_kv)
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\operation.py", line 503, in _validate_and_set_inputs
self.input_spec.validate_inputs(self.name, self.op_type, input_kvs)
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\input_type.py", line 163, in validate_inputs
raise ValueError(msg.format(name, var.name, input_type.type_str,
ValueError: Op "pred_reg.1" (op_type: gather) Input indices="anchor_idxs.1" expects tensor or scalar of dtype from type domain ['int32'] but got tensor[is1,fp32]
The text was updated successfully, but these errors were encountered:
import coremltools as ct
from coremltools.converters.mil.mil import types
coreml_model = ct.convert(
traced_model,
convert_to="mlprogram",
inputs=[ct.TensorType(shape=(1, 3, 16, 224, 224),dtype=types.float)]
)
coreml_model.save('yowo_tiny.mlmodel')
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\builder.py", line 168, in _add_op
new_op = op_cls(**kwargs)
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\operation.py", line 190, in init
self._validate_and_set_inputs(input_kv)
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\operation.py", line 503, in _validate_and_set_inputs
self.input_spec.validate_inputs(self.name, self.op_type, input_kvs)
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\input_type.py", line 163, in validate_inputs
raise ValueError(msg.format(name, var.name, input_type.type_str,
ValueError: Op "pred_reg.1" (op_type: gather) Input indices="anchor_idxs.1" expects tensor or scalar of dtype from type domain ['int32'] but got tensor[is1,fp32]
The text was updated successfully, but these errors were encountered: