From b3a2c085315d8551958ab3378fc329331e0c3a82 Mon Sep 17 00:00:00 2001 From: Olivier DEBAUCHE Date: Sat, 25 May 2024 12:17:40 +0200 Subject: [PATCH] Update xception.py --- fl_common/models/xception.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fl_common/models/xception.py b/fl_common/models/xception.py index e034fe8a..2007f47d 100644 --- a/fl_common/models/xception.py +++ b/fl_common/models/xception.py @@ -32,10 +32,10 @@ def get_xception_model(xception_type, num_classes): try: return create_model( - xception_type, pretrained=True, num_classes=num_classes + xception_type, pretrained=True, num_classes=num_classes, ) except RuntimeError as e: print(f"{xception_type} - Error loading pretrained model: {e}") return create_model( - xception_type, pretrained=False, num_classes=num_classes + xception_type, pretrained=False, num_classes=num_classes, )