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 ·
1200 阅读 ·
0 评论