Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

fixed incorrect pycharme-refactoring

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-25 16:28:00 +01:00
parent 6853d862ed
commit 03f269c2d7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/keras_MnWE.py
View File

@ -26,7 +26,7 @@ def make_model():
dense = layers.Dense(64, activation="relu")
x = dense(inputs)
outputs = layers.Dense(10)(x)
model = keras.ServiceEstimator(inputs=inputs, outputs=outputs, name="mnist_model")
model = keras.Model(inputs=inputs, outputs=outputs, name="mnist_model")
model.compile(
loss=keras.losses.SparseCategoricalCrossentropy(from_logits=True),
optimizer=keras.optimizers.RMSprop(),