tensorflow\core\framework\op_kernel.cc:1318] OP_REQUIRES failed at transpose_op.cc:199 :

 

 

这是tensorflow 一个经常性错误,错误的原因在于:显卡内存不够。
解决方法就是降低显卡的使用内存,途径有以下几种措施:
1 减少Batch 的大小

2 分析错误的位置,在哪一层出现显卡不够,比如在全连接层出现的,则降低全连接层的维度,把2048改成1042啥的

3 增加pool 层,降低整个网络的维度。

4 修改输入大小

posted @ 2021-03-25 22:31  菜鸡一枚  阅读(123)  评论(0)    收藏  举报