search

google / in-silico-labeling

Code for In silico labeling: Predicting fluorescent labels in unlabeled images
Apache License 2.0
260 stars 79 forks source link

Run on GPU #5

Closed Irnamosa closed 4 years ago

Irnamosa commented 4 years ago

Hi

Is there any way I could run this repo on GPU instead of CPU for some speed-up?

emchristiansen commented 4 years ago

I suspect you could get it working on GPU by just wrapping the relevant part in a tf.device context: https://www.tensorflow.org/guide/gpu

Otherwise, you might want to find a U-Net variant that's being actively maintained / written using TF 2.0.

Irnamosa commented 4 years ago

Thanks for the help, I managed to get it working!