From e577b6a2b62cb97f85020007c74a3e7fc3f883ba Mon Sep 17 00:00:00 2001 From: Harim Kang Date: Fri, 23 Aug 2024 18:51:01 +0900 Subject: [PATCH] Fix Semi-SL for ViT accuracy drop (#3883) Remove init_args --- src/otx/algo/classification/vit.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/otx/algo/classification/vit.py b/src/otx/algo/classification/vit.py index 46c4e030d8d..55beb76deff 100644 --- a/src/otx/algo/classification/vit.py +++ b/src/otx/algo/classification/vit.py @@ -290,7 +290,6 @@ def _build_model(self, num_classes: int) -> nn.Module: in_channels=vit_backbone.embed_dim, ), loss=nn.CrossEntropyLoss(reduction="none"), - init_cfg=init_cfg, ) return ImageClassifier(