dalinvip / PyTorch_Bert_Text_Classification

PyTorch Bert Text Classification
Apache License 2.0
31 stars 5 forks source link

Bert feature 里面为什么会有 LSTM 结构 #5

Open ume-technology opened 3 years ago

ume-technology commented 3 years ago

如题。 我看到博主用的 LSTM 做分类。 但是我想改成 CNN。调试了一下午,代码中 Bert_feature 中包含了一个 LSTM 结构,是我理解预训练模型有问题?不清楚为什么会这样。这套代码整体看起来总觉得怪怪的。

wplf commented 1 year ago

他可能写错了,model里少写了bert层,一般情况下是bert+lstm一起用的