How do I run the model.py file using GPU?

[danielJang93]

I am using Ubuntu 16.04 and tensorflow version 1.15. I have ran the code using GPU but I says it is out memory. How can I fix this issue?

[simontomaskarlsson]

Hi @danielJang93,

Try using the option of loading the images during training. Enable this on row 75 in the model by
self.use_data_generator = True