Tensorflow report_tensor_allocations_upon_oom

tensorflow_report_tensor_allocations_upon_oom.py

# Working Example to use report_tensor_allocations_upon_oom
# with recent Tensorflow and Keras
    config = tf.ConfigProto()
    config.gpu_options.allow_growth = True
    sess = tf.Session(config = config)
    K.set_session(sess)

    run_opts = tf.RunOptions(report_tensor_allocations_upon_oom = True)
    run_metadata = tf.RunMetadata()
    # ...
    model.compile(optimizer='adam',
              loss=od_loss,
              metrics=metrics,, options=run_opts, run_metadata=run_metadata)