PyTorch是一个基于Torch的 开源机器学习库,用于自然语言处理等应用程序。
腾讯云 · 智能钛产品团队 (已认证)
1.您好,不好意思,目前暂不支持对工作流里的pytorch版本进行升级。
2.您先确认下您创建的notebook是否是带gpu版本的。
3.新建任务需要等待很久,是因为目前集群资源较紧张,需排队等待资源调度。
您应该使用2D张量作为输入,因为BatchNorm1d使用迷你批次:
BatchNorm1d
在2D或3D输入上应用批量标准化(一批1D输入和可选的附加通道尺寸)... 更多
您可以尝试torch.unsqueeze修复输入尺寸。更多
torch.unsqueeze
你好,请问你解决了吗?怎么解决的呢?先谢过
对我来说,我正在复制RNN名称分类示例中的代码。我添加了优化器和标准模式,而示例是手动操作并手动更新权重。我不小心为优化器添加了一个动量值,这就是我的问题所在。将动量设置为0固定它的默认值。
0
扫码关注云+社区
领取腾讯云代金券