首页
学习
活动
专区
工具
TVP
发布

深度学习入门之PyTorch学习笔记#5

RNN学习预测飞机月流量,非常简单,不过自己练习的时候犯了两个错误,一是转Variable的时候放在了循环体内:

另一个是,因为用了gpu版,所有数据都要先.cuda()变成gpu版后才能使用,同时网络输出的结果也是gpu版,这时候直接把测试结果转成numpy用来绘图的时候报错了,需要先转成cpu版

我训练了5000次得到的效果:

  • 发表于:
  • 原文链接http://kuaibao.qq.com/s/20180114G07H9F00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券