Fix Semi-SL for ViT accuracy drop (#3883)

Remove init_args
openvinotoolkit · Aug 23, 2024 · e577b6a · e577b6a
1 parent ccf2d50
commit e577b6a
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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(