fix SGD bug

This commit is contained in:
chenxuanhong
2023-04-25 22:39:49 +08:00
parent dd1ecdd2a7
commit d4bf5f9984
5 changed files with 174 additions and 11 deletions
+4 -4
View File
@@ -1,4 +1,4 @@
from .models import ArcMarginModel
from .models import ResNet
from .models import IRBlock
from .models import SEBlock
from .arcface_models import ArcMarginModel
from .arcface_models import ResNet
from .arcface_models import IRBlock
from .arcface_models import SEBlock