import torch
from torch import nn
class LSTM(nn.module):
def __init__(self):
super(LSTM,self).__init__()
self.feature = nn.LSTM(input_size=28,hidden_size=64,num_layers=1)
self.out = nn.Linear(64,10)
def forward(x,self):
feat,(a,b)= self.feature(x.permute(1,0,2),None)
return self.out(feat[-1,:,:])
net = LSTM().cuda()
我可以在cpu上打印网络的结构,而anaconda返回 (UnicodeDecodeError:utf-8编解码器无法解码位置67处的字节0xb0:无效的起始字节),如果我在gpu上尝试上网。
还有一件事,我可以在我的cuda上运行Lenet,alexnet和vgg。但这不适用于LSTM。
如何演奏?