Обучение на графическом процессоре происходит слишком медленно [закрыто]Python

Программы на Python
Anonymous
Обучение на графическом процессоре происходит слишком медленно [закрыто]

Сообщение Anonymous »


Я построил глубокую нейронную сеть со слоями ConvLSTM1D, используя библиотеку Keras. Сеть не такая большая, но на обучение модели уходит слишком много времени.

Количество обучаемых параметров — всего 4893.

Вот краткое описание модели.

Модель: "model_4" _________________________________________________________________ Слой (тип) Выходная форма Параметр # =============================================== =============== input_7 (InputLayer) [(Нет, 256, 1, 1)] 0 _________________________________________________________________ conv_lst_m1d_13 (ConvLSTM1D) (нет, 256, 1, 8) 608 _________________________________________________________________ conv_lst_m1d_14 (ConvLSTM1D) (нет, 256, 1, 8) 1056 _________________________________________________________________ conv_lst_m1d_15 (ConvLSTM1D) (нет, 1, 16) 3136 _________________________________________________________________ Flatten_4 (Сгладить) (Нет, 16) 0 _________________________________________________________________ Density_12 (Плотный) (Нет, 4) 68 _________________________________________________________________ Density_13 (Плотный) (Нет, 4) 20 _________________________________________________________________ Density_14 (Плотный) (Нет, 1) 5 =============================================== =============== Всего параметров: 4893 Обучаемые параметры: 4893 Необучаемые параметры: 0 Я обучил модель LSTM, все в порядке.

Вернуться в «Python»