摘要: loss的获取 在看别人代码的时候发现都是 在想为什么不直接使用loss呢,因为pytorch使用Variable跟踪变量(4.0后合并为Tensor),也就是直接使用loss,那么pytorch认为其还在参与运算,其在一个batch后依旧存在于网络中而不是释放掉,所以资源占用会越来越大。 最新版本 阅读全文
posted @ 2018-11-01 21:45 Luke_Ye 阅读(574) 评论(0) 推荐(0) 编辑
摘要: numpy数组增加维度 比如二维灰度图片转化成三维带channel的数组 阅读全文
posted @ 2018-11-01 20:02 Luke_Ye 阅读(175) 评论(0) 推荐(0) 编辑