execution_mode argument raises ArgumentError

ankane / onnxruntime-ruby

Run ONNX models in Ruby

MIT License

111 stars 5 forks source link

Closed lalusaud closed 3 years ago

lalusaud commented 3 years ago

Hey @ankane, thank you for this awesome gem! It works perfect with all the other session options except execution_mode.

model = OnnxRuntime::Model.new('model.onnx', { execution_mode: :sequential })

For some reason, this fails with error message:

ArgumentError (wrong number of arguments (2 for 0))

Is that a bug? I tried to look into it but seems like it is linked to FFI, which is out of my skills at the moment.

ankane commented 3 years ago

Hey @lalusaud, thanks for reporting! Fixed on master.

lalusaud commented 3 years ago

Works now on master! Thank you