From 43ec7d89acf0070cb072099a676c5fb3b2eb6e50 Mon Sep 17 00:00:00 2001
From: Tianhao Wu <90681352+TianhaoTonyWu@users.noreply.github.com>
Date: Tue, 31 Mar 2026 15:10:55 +0800
Subject: [PATCH] Rename method from get_vision_embed to get_image_embed

Fix the inconsistency between syncnet.py and train.py when adapted weight is enabled
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 2c9f0ecd..e82dbf23 100755
--- a/train.py
+++ b/train.py
@@ -229,7 +229,7 @@ def get_ganloss_weight(step):
                     
                     # Calculate adapted weight based on audio-visual similarity
                     if cfg.use_adapted_weight:
-                        vision_embed_gt = syncnet.get_vision_embed(gt_frames)
+                        vision_embed_gt = syncnet.get_image_embed(gt_frames)
                         image_audio_sim_gt = F.cosine_similarity(
                             audio_embed, 
                             vision_embed_gt,