Update xception.py

Smartappli · May 25, 2024 · b3a2c08 · b3a2c08
1 parent f4ae7bb
commit b3a2c08
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,
         )