ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[45,28,28,512]
Mikoto10032 opened this issue · comments
OK，finally I fix this problem.I decrease the batchsize of the network, then it really works.
hizhangp / yolo_tensorflow
Tensorflow implementation of YOLO, including training and test phase.