当我使用 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 进行了实验。
我想知道如何加快训练过程,或者我的配置是否有任何问题,谢谢。