how to use only partial GPU memory

Borda / keras-yolo3

A Keras implementation of YOLOv3 (Tensorflow backend) a successor of qqwweee/keras-yolo3

MIT License

31 stars 10 forks source link

Closed Borda closed 5 years ago

Borda commented 5 years ago

While prediction the model (even Tiny version) allocates complete GPU memory...