thereblue
thereblue
2021-01-25 23:45

关于使用pytorch构建GRU

50
  • python

我使用了pytroch实现了GRU,调用了原有的模型,然后先是报了输入数据缺少维度的问题,当我使用

X_train = X_train.reshape( X_train.shape[0], 1, X_train.shape[1])之后,在loss函数却出现了ValueError: Expected target size (128, 10), got torch.Size([128])的报错,

我不清楚为什么会有这个问题,因为原本的BPNN的代码中也是这样使用的loss函数,没有问题,这里的loss函数使用的是交叉熵损失

loss_function = torch.nn.CrossEntropyLoss(),我也知道这个错误是维度的问题,可是怎么改,我毫无头绪,请大佬指点

函数

  • 点赞
  • 回答
  • 收藏
  • 复制链接分享

8条回答