Блог

Я обучил модель с использованием TensorFlow и пытаюсь загрузить ее в скрипт python на своем RPi 4 с помощью load_model() . Когда я пытаюсь это сделать, я получаю следующую ошибку от модели: выделение 120000 превышает 10% свободной системной памяти. Я почти уверен, что моя модель не займет так много места (у меня карта на 32 ГБ), но почему происходит эта ошибка и как я могу ее исправить?

Я читал другие темы и проблему с GitHub, но проблема решается только тогда, когда я создаю форму ввода (1,1,3) и удаляю все слои свертки из своего CNN. Однако это не может сработать для моего проекта.

Ниже приводится краткое изложение моей модели тензорного потока:

 Model: "sequential_53"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
reshape_53 (Reshape) (None, 60, 20, 1) 0
_________________________________________________________________
flatten_53 (Flatten) (None, 1200) 0
_________________________________________________________________
dense_155 (Dense) (None, 25) 30025
_________________________________________________________________
dense_156 (Dense) (None, 25) 650
_________________________________________________________________
dense_157 (Dense) (None, 1) 26
=================================================================
Total params: 30,701
Trainable params: 30,701
Non-trainable params: 0
_________________________________________________________________