Я пытаюсь заставить 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.
Как можно преобразовать веса? Кто-нибудь может привести пример, когда веса могут быть преобразованы или загружены правильно?