Error with load_learner

Part 1 (2020)
1

Hello,

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

image
image1008×453 55.1 KB

The code and export.pkl are both available on:

2

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.

3

Thanks, it worked!

1 Like