search

jocicmarko / ultrasound-nerve-segmentation

Deep Learning Tutorial for Kaggle Ultrasound Nerve Segmentation competition, using Keras
MIT License
939 stars 329 forks source link

Keras GPU and CPU #76

Open mxl94213 opened 6 years ago

mxl94213 commented 6 years ago

I used the tensorflow as the backend. However , my computer doesn't have a GPU . If i want to use the CPU to train the model, what should i modify the file "train.py"?

mxl94213 commented 6 years ago

@jocicmarko