随笔分类 - Bug记录
摘要:解决: pip instal scikit-image==0.19.3
阅读全文
摘要:1. SKNet输入要求x.shape=[2,3,256,256] 直接concat[input, V_chanel] shape=[2,4,256,256] 报错显示: 解决办法:conv(4,3,3,3)通过out_chanel=3强制转换为3通道 2.测试epoch=10,ssim=0.1,
阅读全文
摘要:1. 获取GPU model.to('cuda:0') 2. 代码跑起来之后中途卡死:调用到其他显卡 则会卡死 3. 报错:RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED (1)检查是否放在GUP上 (2)batch_size. num
阅读全文