2019年9月3日
can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
摘要： predict=predict.data.numpy() 这一行报错意思是:如果想把CUDA tensor格式的数据改成numpy时，需要先将其转换成cpu float-tensor随后再转到numpy格式。 numpy不能读取CUDA tensor 需要将它转化为 CPU tensor将predi 阅读全文
posted @ 2019-09-03 17:23 fuge92 阅读(1745) 评论(0) 推荐(0)
RuntimeError: Expected object of backend CUDA but got backend CPU for argument #4 'mat1'
摘要： 原因：变量没有加cuda data=data.cuda() 阅读全文
posted @ 2019-09-03 17:19 fuge92 阅读(6507) 评论(0) 推荐(0)
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
摘要： 将loss.data[0] 改为loss.item() 阅读全文
posted @ 2019-09-03 17:05 fuge92 阅读(2266) 评论(0) 推荐(1)
参数更新
摘要： 1、参数更新 SGD、Momentum、AdaGrad(学习率衰减）、Adam 2、权重初始值 Xavier初始值（sigmoid、tanh）、He初始值（relu) 3、Batch Normalization 4、正则化 5、权重衰减 6、Dropout 7、超参数（贝叶斯最优化） 设定超参数的范 阅读全文
posted @ 2019-09-03 10:47 fuge92 阅读(289) 评论(0) 推荐(0)