Fixed bug on leras

This commit is contained in:
Cioscos 2021-10-28 11:30:18 +02:00
commit d15fe9ebaf

View file

@ -107,7 +107,7 @@ class nn():
else:
nn.tf_default_device_name = f'/{device_config.devices[0].tf_dev_type}:0'
config = tf.ConfigProto()
config = tf.ConfigProto(allow_soft_placement=True)
config.gpu_options.visible_device_list = ','.join([str(device.index) for device in device_config.devices])
config.gpu_options.force_gpu_compatible = True