Доступ к параметрам сверточной нейронной сети - PullRequest
0 голосов
/ 06 мая 2020

Я пытаюсь переключиться с pytorch на тензорный поток, и, поскольку модель теперь кажется фиксированной вещью в тензорном потоке, я сталкиваюсь с проблемой при работе со сверточными нейронными сетями. только один слой Conv1D и ядро ​​с размером 2. Я хочу обучить его на небольшой конфигурации, скажем, с размером ввода 16, а затем экспортировать результаты обучения с размером ввода 32.

Как я могу получить доступ к 3 параметрам в этой сети? (2 ядра, 1 смещение) Я хочу применить их для случая большего размера. Я борюсь, потому что мне нужно предварительно определить размер ввода модели, в случае с pytorch этого не было.

Спасибо за ответ, я нашел только устаревшие ответы на этот вопрос

1 Ответ

2 голосов
/ 06 мая 2020

model.layers[0].get_weights() дает веса первого слоя, предполагая, что model является объектом tf.keras.Model.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...