Проблема с «IndexError: _Map_base::at» при преобразовании модели тензорного потока в тензорный поток облегченной версии. ⇐ Python

[Anonymous]

Я хочу протестировать производительность квантованной модели тензорного потока. Сначала мне нужно квантовать модель с помощью tensorflow lite, и я получил модель .tflite.
Когда я конвертирую тензорный поток модель .h5 в .tflite, распечатайте ошибку:

Произошло исключение: IndexError
_Map_base::at
File
tflite_model = Converter.convert()
IndexError: _Map_base::at"

Версия тензорного потока — V2.4; Ubuntu 22.04; RX3090
Код выглядит следующим образом:
исходный код
if __name__ == '__main__':

dpcrn = DPCRN_model(batch_size = 1,
length_in_s = 5,
lr = 1e-3,
block_len = 400,
block_shift = 200)
my_model = dpcrn.build_DPCRN_model()
converter = tf.lite.TFLiteConverter.from_keras_model(my_model)
converter.optimizations = [tf.lite.Optimize.DEFAULT]
converter.target_spec.supported_ops = [tf.lite.OpsSet.SELECT_TF_OPS,tf.lite.OpsSet.TFLITE_BUILTINS_INT8] #
converter.representative_dataset = representative_data_gen
tflite_model = converter.convert()

quant_model_path = './model_tflite.tflite'
with open(quant_model_path, 'wb') as f:
f.write(tflite_model)


Подробнее здесь: https://stackoverflow.com/questions/790 ... -to-tensor