I have trained fastai model and created a pickle file. When I load pickle file for model prediction - everything works fine. I created docker file and tested predictions. Again everything is fine.
But in production I am getting following error -
AttributeError: Can’t get attribute ‘_rebuild_from_type’ on <module ‘fastai.torch_core’ from ‘/opt/conda/lib/python3.7/site-packages/fastai/torch_core.py’>
Details -
Traceback (most recent call last):
File “/opt/conda/lib/python3.7/runpy.py”, line 193, in _run_module_as_main
“main”, mod_spec)
File “/opt/conda/lib/python3.7/runpy.py”, line 85, in _run_code
exec(code, run_globals)
File “/opt/conda/lib/python3.7/site-packages/flask/main.py”, line 3, in
main()
File “/opt/conda/lib/python3.7/site-packages/flask/cli.py”, line 995, in main
cli.main(args=sys.argv[1:])
File “/opt/conda/lib/python3.7/site-packages/flask/cli.py”, line 601, in main
return super().main(*args, **kwargs)
File “/opt/conda/lib/python3.7/site-packages/click/core.py”, line 782, in main
rv = self.invoke(ctx)
File “/opt/conda/lib/python3.7/site-packages/click/core.py”, line 1259, in invoke
return _process_result(sub_ctx.command.invoke(sub_ctx))
File “/opt/conda/lib/python3.7/site-packages/click/core.py”, line 1066, in invoke
return ctx.invoke(self.callback, **ctx.params)
File “/opt/conda/lib/python3.7/site-packages/click/core.py”, line 610, in invoke
return callback(*args, **kwargs)
File “/opt/conda/lib/python3.7/site-packages/click/decorators.py”, line 73, in new_func
return ctx.invoke(f, obj, *args, **kwargs)
File “/opt/conda/lib/python3.7/site-packages/click/core.py”, line 610, in invoke
return callback(*args, **kwargs)
File “/opt/conda/lib/python3.7/site-packages/flask/cli.py”, line 850, in run_command
app = DispatchingApp(info.load_app, use_eager_loading=eager_loading)
File “/opt/conda/lib/python3.7/site-packages/flask/cli.py”, line 325, in init
self._load_unlocked()
File “/opt/conda/lib/python3.7/site-packages/flask/cli.py”, line 351, in _load_unlocked
self._app = rv = self.loader()
File “/opt/conda/lib/python3.7/site-packages/flask/cli.py”, line 411, in load_app
app = locate_app(self, import_name, None, raise_if_not_found=False)
File “/opt/conda/lib/python3.7/site-packages/flask/cli.py”, line 260, in locate_app
import(module_name)
File “/workspace/app.py”, line 8, in
learner = load_learner(’./uniform_fastai.pkl’)
File “/opt/conda/lib/python3.7/site-packages/fastai/learner.py”, line 374, in load_learner
res = torch.load(fname, map_location=‘cpu’ if cpu else None, pickle_module=pickle_module)
File “/opt/conda/lib/python3.7/site-packages/torch/serialization.py”, line 594, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File “/opt/conda/lib/python3.7/site-packages/torch/serialization.py”, line 853, in _load
result = unpickler.load()
AttributeError: Can’t get attribute ‘_rebuild_from_type’ on <module ‘fastai.torch_core’ from ‘/opt/conda/lib/python3.7/site-packages/fastai/torch_core.py’>