keras训练过程中发生的一些报错及其解决办法
ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[32,x,x,x]
shape[0]为batch_size, 这个错误是batch_size过大，造成gpu内存超出，适当降低batch_size即可
keras指定gpu
import kera...
原创
2019-11-27 09:53:29 ·
1270 阅读 ·
0 评论
关注
分享
扫一扫