xuannianz / EfficientDet

EfficientDet (Scalable and Efficient Object Detection) implementation in Keras and Tensorflow
Apache License 2.0
1.39k stars 395 forks source link

GPU usage is very low #238

Closed rajan780 closed 3 years ago

rajan780 commented 3 years ago

Hi all I am able to train but gpu usage is very low only 584MB / 10988MB. can anyone tell me how to run the model on gpu. and when i am running the train.py with multiprocessing training got stuck. Any reason for that?