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 阅读(1735) 评论(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 阅读(6493) 评论(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 阅读(2228) 评论(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 阅读(276) 评论(0) 推荐(0) 编辑

公告