search

hizhangp / yolo_tensorflow

Tensorflow implementation of YOLO, including training and test phase.
MIT License
795 stars 442 forks source link

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[45,28,28,512] #71

Closed Mikoto10032 closed 6 years ago

Mikoto10032 commented 6 years ago

OK，finally I fix this problem.I decrease the batchsize of the network, then it really works.