fix: altered tf.config function call to be compatible with tf 1.15
@@ -185,7 +185,7 @@ class TFKerasTrialController(det.LoopTrialController):
if len(gpus) > 0:
local_rank = hvd.local_rank() if hvd_config.use else 0
gpu = gpus[local_rank]
- tf.config.set_visible_devices(gpu, "GPU")
+ tf.config.experimental.set_visible_devices(gpu, "GPU")
tf.config.experimental.set_memory_growth(gpu, True)
return None
