From 070749880e7a6bcf35c37b4b98dff78cdbe02461 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Fri, 1 Apr 2022 21:13:15 +0200 Subject: [PATCH] removed obsolete code --- image_prediction/redai_adapter/model_wrapper.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/image_prediction/redai_adapter/model_wrapper.py b/image_prediction/redai_adapter/model_wrapper.py index a90c342..5bb9a70 100644 --- a/image_prediction/redai_adapter/model_wrapper.py +++ b/image_prediction/redai_adapter/model_wrapper.py @@ -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