Closed bdubayah closed 3 years ago
Hi @hannah-rae and @stevenlujpl , the PAE ranking modules are now finalized and ready for review – I edited the top comment to include all the major changes. Let me know if there's any questions or changes needed!
@bdubayah Thanks for making the changes, and they all look great to me.
I still have some questions:
This is how I limited the GPU memory usage in tensorflow 1.x:
import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)
Another option is to set the visible GPUs to be a single GPU.
Good catch, I didn't know that tensorflow did that - it will run on GPUs if available. Maybe we could by default set it to one gpu with growth on and then add config options to allow for multiple gpus and enabling/disabling growth? Or an option to entirely disable GPUs?