7 lines
279 B
Python
7 lines
279 B
Python
import tensorflow as tf
|
|
|
|
interpreter = tf.lite.Interpreter(model_path="best_int8.tflite")
|
|
interpreter.allocate_tensors()
|
|
input_details = interpreter.get_input_details()
|
|
output_details = interpreter.get_output_details()
|
|
print(input_details[0]['dtype'], output_details[0]['dtype']) |