tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[326704,1203] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc

yangxue0827 / FPN_Tensorflow

A Tensorflow implementation of FPN detection framework.

415 stars 150 forks source link

Closed pyupcgithub closed 5 years ago

pyupcgithub commented 5 years ago

Had anyone meet this question ?

the batch_size=1 , how would this happen ?

yangxue0827 commented 5 years ago

give me your cfgs.py @pyupcgithub

yangxue0827 commented 5 years ago

Because some images are too large. Setting small value of short_side_len in cfgs.py or removing large images.