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.
Tensorflow implementation of YOLO, including training and test phase.
Mikoto10032 opened this issue · comments
OK,finally I fix this problem.I decrease the batchsize of the network, then it really works.