diff --git a/run.py b/run.py index 494c17f..2c93545 100644 --- a/run.py +++ b/run.py @@ -67,7 +67,7 @@ with tf.device('/GPU:0'): keras.layers.Dense(1000, activation='relu'), keras.layers.Dense(10, activation='sigmoid') ]) - + # g model.compile(optimizer='SGD', loss='categorical_crossentropy', metrics=['accuracy'])