Керас: CUDNNGRU весит в ГРУ - PullRequest
0 голосов
/ 28 марта 2020

Я пытаюсь заставить https://github.com/YunjaeChoi/vaemols работать. Пользователь использовал слой CuDNNGRU. Я хотел, чтобы он работал на моем p c, где нет доступного графического процессора. Я просто хочу кодировать и декодировать. Я не хочу тренироваться дальше.

Если быть точным: я хочу сделать эту работу https://github.com/YunjaeChoi/vaemols/blob/master/structure_variation.ipynb

Я изменил файл vaemols / vae.py, где Я заменил каждый слой tf.keras.layers.CuDNNGRU на tf.keras.layers.GRU. Кроме того, я добавил параметры для слоя GRU с помощью reset_after=True, recurrent_activation='sigmoid', потому что я прочитал его здесь TensorFlow Keras CuDNNGRU в преобразование GRU .

После загрузки весов контрольных точек полученные результаты кажутся очень неправильными. Похоже, что существует несовместимость между весами CuDNNGRU и GRU.

Как можно преобразовать веса? Кто-нибудь может привести пример, когда веса могут быть преобразованы или загружены правильно?

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