init

ViTAE-Transformer · Jun 8, 2023 · 9765d7b · 9765d7b
1 parent f8e3367
commit 9765d7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MAEPretrain_SceneClassification/vit_win_rvsa_kvdiff.py b/MAEPretrain_SceneClassification/vit_win_rvsa_kvdiff.py
@@ -389,7 +389,7 @@ def forward(self, x, H, W):
             sampling_angle_k = sampling_angle_k.reshape(num_predict_total, 1, window_num_h, window_num_w)
 
             sampling_angle_v = self.sampling_angles_v(x)
-            sampling_angle_v = sampling_angle_k.reshape(num_predict_total, 1, window_num_h, window_num_w)
+            sampling_angle_v = sampling_angle_v.reshape(num_predict_total, 1, window_num_h, window_num_w)
 
             # first scale