google-research / evoflow

Apache License 2.0
32 stars 10 forks source link

Use CPU to perform small problem as the cost to go to GPU is not worth it #45

Open ebursztein opened 4 years ago

ebursztein commented 4 years ago

Use the shape inference and param counts to decide if we should use TF.CPU or tf.GPU