home adaptive_computation ponder_net
View code on Github
这是 P onderNet:学会思考论文的 PyTorch 实现。
PonderNet 根据输入调整计算。它会根据输入更改循环网络上要执行的步骤数。PonderNet 通过端到端梯度下降来学习这一点。