temp

openvinotoolkit · Nov 1, 2023 · 2ac7f25 · 2ac7f25
1 parent f350ed2
commit 2ac7f25
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -10,7 +10,6 @@ All notable changes to this project will be documented in this file.
 - Add Anomaly modelAPI changes (<https://github.com/openvinotoolkit/training_extensions/pull/2563>)
 - Fix IBLoss enablement with DeiT-Tiny when class incremental training (<https://github.com/openvinotoolkit/training_extensions/pull/2595>)
 
-
 ## \[v1.4.3\]
 
 ### Enhancements

diff --git a/...x/algorithms/classification/adapters/mmcls/models/heads/custom_vision_transformer_head.py b/...x/algorithms/classification/adapters/mmcls/models/heads/custom_vision_transformer_head.py
@@ -31,7 +31,7 @@ def loss(self, cls_score, gt_label, feature=None):
             losses["accuracy"] = {f"top-{k}": a for k, a in zip(self.topk, acc)}
         losses["loss"] = loss
         return losses
-    
+
     def forward_train(self, x, gt_label, **kwargs):
         """Forward_train fuction of CustomVisionTransformerClsHead class."""
         x = self.pre_logits(x)