![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
North槛内人
这个作者很懒,什么都没留下…
展开
-
Python将字符串变为原始字符串 repr()函数
问题:从一个路径读取文件时,由于输入的路径字符串为单斜线格式,导致转义后不能正确代表路径,从而读取错误:例如:原始路径字符串:filename = 'H:\Head Pose Dataset\csvs\face.png'用cv读取该路径图片显然会出错。两种方法:r'H:\Head Pose Dataset\csvs\face.png'或者:repr(filena...原创 2019-11-10 10:52:10 · 4817 阅读 · 0 评论 -
解决pytorch训练时正常测试时爆显存的问题
问题:Runtime Error: CUDA out of memory.解决方案:1.降低batch_size, 有时候管用,有时候不管用2. 测试时,with torch.no_grad()def early_test(self): with torch.no_grad(): self.model.eval() for path i...原创 2019-10-28 20:09:51 · 3348 阅读 · 0 评论