search

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
492 stars 106 forks source link

dlr.dlr_model.DLRError: TVMError: Check failed: read_size == expected_size #287

Closed YingUK closed 4 years ago

YingUK commented 4 years ago

Hi,

I follow this tutorial : https://docs.amazonaws.cn/en_us/greengrass/latest/developerguide/ml-dlc-console.html. And using Jetson Nano. The greengrass is installed correctly, and dlr and opencv are built from source using cmake. The local lambda function can be triggered, but I saw below error in the lambda function log. could anyone help with this issue?

[2020-11-17T14:34:05.231Z][ERROR]- return self._impl.run(input_values) [2020-11-17T14:34:05.231Z][ERROR]- File “/usr/local/lib/python3.7/dist-packages/dlr-1.5.0-py3.7.egg/dlr/dlr_model.py”, line 466, in run [2020-11-17T14:34:05.231Z][ERROR]- self._set_input(self.input_names[0], input_values) [2020-11-17T14:34:05.231Z][ERROR]- File “/usr/local/lib/python3.7/dist-packages/dlr-1.5.0-py3.7.egg/dlr/dlr_model.py”, line 353, in _set_input [2020-11-17T14:34:05.231Z][ERROR]- c_int(len(shape)))) [2020-11-17T14:34:05.231Z][ERROR]- File “/usr/local/lib/python3.7/dist-packages/dlr-1.5.0-py3.7.egg/dlr/dlr_model.py”, line 185, in _check_call [2020-11-17T14:34:05.231Z][ERROR]- raise DLRError(self._lib.DLRGetLastError().decode(‘ascii’)) [2020-11-17T14:34:05.231Z][ERROR]-dlr.dlr_model.DLRError: TVMError: Check failed: read_size == expected_size (5760000 vs. 150528) : Mismatch found in input data size. Value read: 5760000, Expected: 150528 [2020-11-17T14:34:05.231Z][ERROR]-Stack trace: [2020-11-17T14:34:05.231Z][ERROR]- File “/home/nvidia/neo-ai-dlr/src/dlr_tvm.cc”, line 172 [2020-11-17T14:34:05.231Z][ERROR]- [bt] (0) /usr/local/lib/python3.7/dist-packages/dlr-1.5.0-py3.7.egg/dlr/libdlr.so(dmlc::LogMessageFatal::~LogMessageFatal()+0x88) [0x7f65fe8ef0] [2020-11-17T14:34:05.231Z][ERROR]- [bt] (1) /usr/local/lib/python3.7/dist-packages/dlr-1.5.0-py3.7.egg/dlr/libdlr.so(dlr::TVMModel::SetInput(char const, long const, void const*, int)+0x488) [0x7f66048ea8] [2020-11-17T14:34:05.231Z][ERROR]- [bt] (2) /usr/local/lib/python3.7/dist-packages/dlr-1.5.0-py3.7.egg/dlr/libdlr.so(SetDLRInput+0x20) [0x7f65ffca50]

trevor-m commented 4 years ago

Hi @YingUK The error Check failed: read_size == expected_size (5760000 vs. 150528) : Mismatch found in input data size. Value read: 5760000, Expected: 150528 means that the input you gave to the model was not the right size. 150528 is 224 * 224 * 3, so it looks like your model can only accept inputs with h=224, w=224, and c=3.

YingUK commented 4 years ago

Hi @trevor-m , you are right, I didn't resize the image properly in the inference code. Now it's fixed. Thank you!