Pytorch
ciciph
这个作者很懒,什么都没留下…
展开
-
自定义皮尔森损失函数遇到错误:tensorflow.python.framework.errors_impl.InvalidArgumentError
ValueError: Shape must be rank 0 but is rank 1 for '{{node myloss/Fill}} = Fill[T=DT_FLOAT, index_type=DT_INT32](myloss/Fill/dims, myloss/strided_slice)' with input shapes: [2], [1335].原创 2020-10-03 21:22:17 · 1392 阅读 · 0 评论 -
解决win10(2080ti)+CUDA9.2+pytorch-gpu(RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED)
解决win10(2080ti)+CUDA9.2+pytorch-gpu(RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED)问题CUDA9.2+Anaconda5.0+Pytorch1.0.0( py3.7_cuda90_cudnn7_1)安装完成后,不调用GPU跑程序,可以正常运行,当调用cuda()后出错:RuntimeErro...原创 2018-12-11 15:01:55 · 11706 阅读 · 4 评论