thereblue
2021-01-25 23:45关于使用pytorch构建GRU
50我使用了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条回答
为你推荐
- pytorch的MNIST代码中loss输出的疑问
- 深度学习
- 人工智能
- 神经网络
- 机器学习
- 1个回答
- module 'torch.nn' has no attribute 'linear'。
- 深度学习
- python
- 机器学习
- list
- 1个回答
- 用anaconda自己配置了一个pytorch环境,那么我可以把base环境下的依赖库的文件夹复制到pytorch环境下,省去自己安装的步骤么????
- python
- 深度学习
- 机器学习
- 1个回答
- pytorch cuda版运行出错 invalid start byte
- python
- 神经网络
- 2个回答
- 找DAN,DDC,JAN,RTN,simNet,ResNet-50等模型的pytorch框架代码。能找几个是几个。
- 深度学习
- 人工智能
- 神经网络
- python
- 机器学习
- 3个回答
换一换