image_batch, label_batch = next(iter(validation_generator))

interpreter.set_tensor(input_details[0]['index'], image_batch)

interpreter.invoke()

tflite_results = interpreter.get_tensor(output_details[0]['index'])

tf_results = model.predict(image_batch)

for tf_result, tflite_result in zip(tf_results, tflite_results):
    np.testing.assert_almost_equal(tf_result, tflite_result, decimal=5)