- 博客(2)
- 收藏
- 关注
RSS订阅原创 #Tensorflow踩坑记#之二 Can not squeeze dim[3]
ValueError: Can not squeeze dim[3], expected a dimension of 1, got 3 for ‘metrics/accuracy/Squeeze’我自己出现这个错误的原因是因为model.compile 中 loss=‘sparse_categorical_crossentropy’我自己的纠正方案：把 model.compile 中 loss改为 loss=‘mse’...
2020-06-17 17:17:41
1714
1
原创 #Tensorflow踩坑记#之一 00M的出现
#Tensorflow踩坑记#之一 00M的出现Hint: If you want to see a list of allocated tensors when OOM happens我这边出现这个Hint，并终止计算的原因是： batch_size 设置太大，稍微调小点即可正常运行
2020-06-17 17:13:10
1182
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人