Neural_Speed_Reading_via_Skim-RNN_PyTorch:PyTorch实施“通过Skim-RNN进行...

介绍 这是在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