tensorflow2.x

How to get other metrics in Tensorflow 2.0 (not only accuracy)?

by

I am adding another answer because this is the cleanest way in order to compute these metrics correctly on your test set (as of 22nd of March 2020). The first thing you need to do is to create a custom callback, in which you send your test data: import tensorflow as tf from tensorflow.keras.callbacks import … Read more

How to prevent tensorflow from allocating the totality of a GPU memory?

by

You can set the fraction of GPU memory to be allocated when you construct a tf.Session by passing a tf.GPUOptions as part of the optional config argument: # Assume that you have 12GB of GPU memory and want to allocate ~4GB: gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333) sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) The per_process_gpu_memory_fraction acts as a hard upper bound … Read more