Fastai wwf,timm trained (exported)model, loading in sagemaker endpoint error

[m-ali-awan]

I am trying to deploy fastai trained model (build using wwf(walk with fastai), and time), to sagemaker endpoint.
I have also included in required libraries as requirements.txt.
But the model is not bein able to loaded by load_learner:

This is my model-fn

# loads the model into memory from disk and returns it
def model_fn(model_dir):
  logger.info('model_fn')
  path_model = Path(model_dir)
  logger.debug(f'Loading model from path: {str(path_model/EXPORT_MODEL_NAME)}')
  print(path_model/EXPORT_MODEL_NAME)
  defaults.device = torch.device('cpu')
  print("Trying to Load Model::::")
  learn = load_learner(path_model/EXPORT_MODEL_NAME, cpu=True)
  print('MODEL-LOADED')
  logger.info('model loaded successfully')
  return learn

So, in the screenshot it can be seen that "MODEL-LOADED" is not printed.
Comparing, for a simple fastai-trained model, as a demo, that is working.

So kindly help me with this.

Thanks alot.

[amaarora]

I think asking this in the walkwithfastai repo might be better suited. Believe this is something Zach has worked on. But, best to ask :)