文件名称:Neural_Speed_Reading_via_Skim-RNN_PyTorch:PyTorch实施“通过Skim-RNN进行神经速度读取”
文件大小:454KB
文件格式:ZIP
更新时间:2024-05-31 23:42:26
pytorch skim-rnn Python
介绍 这是在ICLR 2018上发布的进行的PyTorch实现。 默认情况下,将使用imdb数据集并将其存储在./data文件夹中。 此外,还使用了经过8400训练的300维GloVe词嵌入。 与目标是离散的或不同,Skim RNN引入了Gumbel-softmax参数化技巧,该技巧使撇取目标变得可区分: 用法 python main.py [arguments] 争论 -h, help help -large_cell_size size of the large LSTM -small_cell_size size ofthe small LSTM
【文件预览】:
Neural_Speed_Reading_via_Skim-RNN_PyTorch-master
----skim_rnn.py(4KB)
----.DS_Store(8KB)
----.gitattributes(66B)
----main.py(7KB)
----skim_rnn.png(186KB)
----dataset.py(4KB)
----.idea()
--------misc.xml(185B)
--------workspace.xml(18KB)
--------Skim_RNN_PyTorch.iml(492B)
--------modules.xml(284B)
--------vcs.xml(180B)
----LICENSE(1KB)
----README.md(877B)
----.data()
--------imdb()
----.gitignore(1KB)
----gumbel_softmax.png(15KB)