为什么 TextCNN 运行这么慢?

问题描述 投票:0回答:0

当我使用 TextCNN 进行文本分类时,我发现对于相同的输入,它比 LSTM 慢了将近 10 倍。这是正常的吗?将 mini-batch 设置为 32,训练 CNN 100 batch 需要一个多小时,而 LSTM 只需不到十分钟。我使用了三个卷积核进行卷积运算,比如(2, 768)(3,768)(4,768),输出通道为128。我使用PyTorch作为架构的框架。

self.convs = nn.ModuleList(
            [nn.Conv2d(1, config.num_filters, (k, config.hidden_size)) for k in config.filter_sizes])

今天在使用ldconfig命令的时候遇到了一个错误。这会有什么影响吗?

ldconfig: /lib64/libcudnn_cnn_train.so.8 is not a symbolic link
ldconfig: /lib64/libcudnn_cnn_infer.so.8 is not a symbolic link

我使用 3090 进行了实验。

我想知道如何加快训练过程,或者我的配置是否有任何问题,谢谢。

pytorch conv-neural-network lstm text-classification
© www.soinside.com 2019 - 2024. All rights reserved.