随笔分类 -  pytorch

pytorch显存越来越多的一个自己没注意的原因
摘要：参考：https://blog.csdn.net/qq_27292549/article/details/80250031 我和博主犯了一毛一样的低级错误。。。。 下面是原博解释： 运行着就发现显存炸了 观察了一下发现随着每个batch显存消耗在不断增大.. 参考了别人的代码发现那句loss一般是这 阅读全文

posted @ 2019-01-07 17:46 文狸大人 阅读(10905) 评论(0) 推荐(1)

x = x.view(x.size(0), -1) & x = x.view(-1, num_flat_feature) 的理解
摘要：部分转自于： https://blog.csdn.net/whut_ldz/article/details/78882532 之前对于pytorch的网络编程学习都是大致理解每一层的概念，有些语法语句没有从原理上弄清楚，就比如标题的x = x.view(x.size(0), -1) 。 这句话一般出 阅读全文

posted @ 2018-08-13 16:30 文狸大人 阅读(1585) 评论(0) 推荐(0)

导航