![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
踩坑记
sinat_24673397
这个作者很懒,什么都没留下…
展开
-
#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 · 1681 阅读 · 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 · 1167 阅读 · 0 评论