search

tanaka504 / Multi-turn_Response_Generation

0 stars 0 forks source link

optimizer error when multi-turn training #7

Closed tanaka504 closed 5 years ago

tanaka504 commented 5 years ago

症状

訓練時Embedding層の重みを固定すると発話単位のGRUの更新時にエラーが出る．

解決策

発話単位のEncoderをEmbeddingのクラスとGRUのクラスに分割する

tanaka504 commented 5 years ago

https://discuss.pytorch.org/t/freeze-the-learnable-parameters-of-resnet-and-attach-it-to-a-new-network/949/9 これで解決できた．