This message was deleted.

Hi Shiva,

predict

in

runner.predict.run

means the runner will try to call

predict

method on the original model. Does your model have

predict

method? If you want to call

__call__

method on your model, you can simply write

runner.run(...)