ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 09:07:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix keras_tuner_cifar example raises DeprecationWarning and ValueError

Browse Source
This commit is contained in:
allegroai 2022-07-21 17:23:19 +03:00
parent 157df5dad3
commit 51f512889f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/frameworks/kerastuner/keras_tuner_cifar.py
View File

@ -1,6 +1,6 @@
"""Keras Tuner CIFAR10 example for the TensorFlow blog post.""" """Keras Tuner CIFAR10 example for the TensorFlow blog post."""
import kerastuner as kt import keras_tuner as kt
import tensorflow as tf import tensorflow as tf
import tensorflow_datasets as tfds import tensorflow_datasets as tfds
from clearml.external.kerastuner import ClearmlTunerLogger from clearml.external.kerastuner import ClearmlTunerLogger
@ -9,6 +9,7 @@ from clearml import Task
physical_devices = tf.config.list_physical_devices('GPU') physical_devices = tf.config.list_physical_devices('GPU')
if physical_devices: if physical_devices:
tf.config.experimental.set_visible_devices(physical_devices[0], 'GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True) tf.config.experimental.set_memory_growth(physical_devices[0], True)