From 90d2546fdf3602d69e42ad3aa92bb13dee3abaf2 Mon Sep 17 00:00:00 2001 From: AnXiang Date: Mon, 12 Jul 2021 14:32:10 +0800 Subject: [PATCH] fixed bugs --- recognition/arcface_torch/backbones/mobilefacenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recognition/arcface_torch/backbones/mobilefacenet.py b/recognition/arcface_torch/backbones/mobilefacenet.py index 0629992..65fa36b 100644 --- a/recognition/arcface_torch/backbones/mobilefacenet.py +++ b/recognition/arcface_torch/backbones/mobilefacenet.py @@ -86,7 +86,7 @@ class GDC(Module): class MobileFaceNet(Module): - def __init__(self, fp16=False, scale=1, num_features=512): + def __init__(self, fp16=False, num_features=512): super(MobileFaceNet, self).__init__() self.fp16 = fp16 self.layers = nn.Sequential(