Error with load_learner


Following 02_production.ipynb notebook, I export the model and when I load it back with load_learner, I get the following error:

The code and export.pkl are both available on:

You used your custom function get_color when defining your DataLoaders, and when you load your Learner object, it is looking for this function. Try defining the get_color function again in your inference notebook.

Thanks, it worked!

1 Like