ARM-software / ML-zoo

Apache License 2.0
194 stars 51 forks source link

Test code for QAT model #49

Open mychina75 opened 2 years ago

mychina75 commented 2 years ago

Hi, I tried to test the QAT tflite model, but error reported. ValueError: Cannot set tensor: Got value of type FLOAT32 but expected type INT8 for input 0, name: serving_default_input_1

Based on my understanding, input tensor should be 0.~1., but the model need INT8 input. Would you give some introduction about test code for QAT model? Thanks a lot.