home adaptive_computation ponder_net

View code on Github

PonderNet: 熟考することを学ぶ

これは、論文「PonderNet: 熟考を学ぼう」をPyTorchで実装したものです。

PonderNet は入力に基づいて計算を調整します。入力に基づいてリカレントネットワークで実行するステップの数を変更します。PonderNetはこれを端から端までの勾配降下法で学習します

。