How to train the Model with GPU

kentaroy47 / frcnn-from-scratch-with-keras

:collision:Faster R-CNN from scratch written with Keras

Apache License 2.0

168 stars 106 forks source link

Closed ani16 closed 4 years ago

ani16 commented 4 years ago

The model is successfully running on cpu , but how to run the same model on GPU .I have Nvidia K40.

Secondly, input weight file can be used to place pretrained model? For testing the model for voc , I just need to load the trained model weight ?

kentaroy47 commented 4 years ago

keras should run GPU when it is available. does it grab GPU when training for tutorials like MNIST?

I just need to load the trained model weight ? you should just load the trained weight.