将sentiment分类demo中的lstmemory换成simple_lstm后发现训练速度变慢,并且cost值下降也很慢
Created by: 333caowei
使用sentiment分类demo中的三层LSTM,demo中原始使用的是lstmemory,在文档中发现lstmemory需要配上fc_layer才是完整的lstm layer,于是我替换成了simple_lstm进行尝试,结果发现训练速度变得非常慢,并且cost值变化也很慢。
如图,注释掉的是原始demo中的lstmemory,红框部分是我改写的simple_lstm,并且发现使用simple_lstm后内存为7.4g(lstmemory内存仅为3.8g):