neo-ai / neo-ai-dlr

Neo-AI-DLR is a common runtime for machine learning models compiled by AWS SageMaker Neo, TVM, or TreeLite.
Apache License 2.0
491 stars 107 forks source link

No valid Treelite model files found under folder (XGBOOST) #94

Open rymoore opened 4 years ago

rymoore commented 4 years ago

I've created a Sagemaker XGBOOST model and compiled it for NEO using:

compiled_model = xgb.compile_model(target_instance_family='rasp3b', 
                                   role=role,
                                   framework='xgboost',
                                   framework_version='0.7',
                                   output_path=output_path)

I am deploying this model to a RPI3 via Greengrass for inference using dlr. When I use the following Python code on the device:

model = DLRModel('model', input_shape, output_shape, device)

I get an error "No valid Treelite model files found under folder". My "model" directory has a single file called "compiled_model.so".

What steps am I missing?

hcho3 commented 4 years ago

@rymoore Which version of DLR did you use, and how did you install it?