removed obsolete code
This commit is contained in:
parent
94783c54f2
commit
070749880e
@ -66,10 +66,6 @@ class EfficientNetWrapper:
|
||||
|
||||
model = tf.keras.models.Model(inputs=input_img, outputs=finetuned)
|
||||
|
||||
model.compile(
|
||||
loss="categorical_crossentropy",
|
||||
optimizer="adam",
|
||||
metrics=[tf.keras.metrics.Recall(), tf.keras.metrics.Precision()],
|
||||
)
|
||||
model.compile()
|
||||
|
||||
return model
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user