From d920cce12c870e91d3ff21ab574f961d8555c283 Mon Sep 17 00:00:00 2001 From: LashaO Date: Thu, 13 Jun 2024 19:06:51 +0400 Subject: [PATCH] fixes linting --- wbia/algo/detect/efficientnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wbia/algo/detect/efficientnet.py b/wbia/algo/detect/efficientnet.py index 42d2c476d..19ce53e3f 100644 --- a/wbia/algo/detect/efficientnet.py +++ b/wbia/algo/detect/efficientnet.py @@ -169,7 +169,7 @@ def __init__(self, n_class, model_arch='tf_efficientnet_b4_ns', pretrained=False if n_class is not None: n_features = self.model.classifier.in_features self.model.classifier = nn.Linear(n_features, n_class) - + else: self.model.classifier = nn.Identity(n_features, n_class) ''' @@ -179,7 +179,7 @@ def __init__(self, n_class, model_arch='tf_efficientnet_b4_ns', pretrained=False nn.Linear(n_features, n_class, bias=True) ) ''' - + def process_row(self, row_labels, preds, sort_weights, labels): multi_labels = labels[row_labels.astype(bool)] preds = preds[row_labels.astype(bool)]