spacecruft
/
witzit
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

black for py PEP8 formatting

main
jebba 2022-01-24 14:42:39 -07:00
parent b7916b8dae
commit 55cb751afb
3 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements-dev.txt 100644
View File

@ -0,0 +1 @@
black

4
requirements.txt
View File

@ -1,3 +1,5 @@
#tensorflow==2.7.0
tensorflow_cpu==2.7.0
#tensorflow_gpu==2.7.0
#tensorflow==2.7.0
matplotlib==3.5.1

3
wz-load
View File

@ -29,6 +29,7 @@ import numpy as np
import datetime
import tensorflow as tf
import tensorflow.python.keras
import matplotlib.pyplot as plt
from tensorflow.keras.layers import Dense, Conv2D, Flatten, Dropout, MaxPooling2D
from tensorflow.python.keras import optimizers
from tensorflow.python.keras import Sequential
@ -41,6 +42,8 @@ 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.callbacks import ModelCheckpoint
print("Tensorflow Version: ", tf.__version__)
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))