工作笔记
平时工作中遇到的问题记录
wangyi4422
这个作者很懒,什么都没留下…
展开
-
module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them
问题:module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu当在pytorch下训练好模型之后,测试的时候调用checkpoint,用cpu跑的时候出现这个错误。分析: 出现这个错误主要是因为模型在创建的时候用的...原创 2019-12-13 13:43:12 · 6246 阅读 · 0 评论 -
ios 操作图像imageOrientation的相关问题
这里要讲一个大坑。最近在做ios端的深度学习,从视频中获取图像数据,传入深度学习模型中进行预测,但总是无结果。将图像显示到uiimageview上看是正常的,但是在输入到模型时,旋转了90度,距离砸键盘只差一个小时的距离。终于,在看了这篇博客之后, 茅塞顿开。原来在我输入模型之前,对image进行了像素操作,去掉了图像alpha通道,所以原本图像EXIF中的imageOrientati...原创 2019-10-12 16:37:44 · 584 阅读 · 0 评论 -
Include of non-modular header inside framework module
The real answer is that the location of the imports needs to be changed by the library owner. Those files ifaddrs.h, arpa/inet.h, sys/types.h are getting imported in a .h file in a framework, which Xc...原创 2019-10-11 08:42:44 · 1106 阅读 · 0 评论