From 2d0abfe4f1782f16275c0b975c4e9177994b5abd Mon Sep 17 00:00:00 2001 From: jebba Date: Mon, 24 Jan 2022 14:43:18 -0700 Subject: [PATCH] minor formatting --- wz-load | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/wz-load b/wz-load index 0bd3356..c04ce90 100755 --- a/wz-load +++ b/wz-load @@ -42,11 +42,11 @@ from tensorflow.python.keras.preprocessing import image from tensorflow.python.keras.preprocessing.image import img_to_array from tensorflow.python.keras.preprocessing.image import ImageDataGenerator from tensorflow.python.keras.preprocessing.image import load_img -#from tensorflow.python.keras.utils import plot_model + +# from tensorflow.python.keras.utils import plot_model from tensorflow.python.keras.callbacks import ModelCheckpoint print("Tensorflow Version: ", tf.__version__) -print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU'))) -print("Num CPUs Available: ", len(tf.config.experimental.list_physical_devices('CPU'))) +print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices("GPU"))) +print("Num CPUs Available: ", len(tf.config.experimental.list_physical_devices("CPU"))) print(tf.config.experimental.list_physical_devices()) -