环境:cuda10.0,python3.7,torch 1.2.0

问题1:前向传播时报错:

RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

原因是nn.LSTM这个模块不支持torch.float64,如果直接用torch.from_numpy转换向量,其类型是torch.float64,需要加一个.float()转换为torch.float32

问题2:我的输入是(batch, 7, 5)keras.LSTM直接输出了(batch,50)
原因是keras.LSTM默认输出最后一个时间步的结果,如果return_sequences设置为True,则会输出(batch,7,50)

持续更新中。。。

更多推荐