From 733ab8da1595dc9854a41eee45414eba183641e1 Mon Sep 17 00:00:00 2001
From: Saarthak Kapse <44542181+saarthak-kapse@users.noreply.github.com>
Date: Sat, 8 Feb 2025 00:01:06 -0500
Subject: [PATCH] Update VimB.yaml

---
 imagenet_classification/config/VimB.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imagenet_classification/config/VimB.yaml b/imagenet_classification/config/VimB.yaml
index 4f945e7..08d0719 100644
--- a/imagenet_classification/config/VimB.yaml
+++ b/imagenet_classification/config/VimB.yaml
@@ -20,7 +20,7 @@ warmup_epochs: 5
 num_classes: 1000
 
 vim_config:
-  _target_: models.fastvim.vim_base_patch16_224_final_pool_mean_abs_pos_embed_with_midclstok_div2
+  _target_: models.vim.vim_base_patch16_224_final_pool_mean_abs_pos_embed_with_midclstok_div2
   pretrained: false
   num_classes: ${num_classes}
   drop_path_rate: ${drop_path_rate}