CMG icon indicating copy to clipboard operation
CMG copied to clipboard

embedding updated in MM_EMA

Open RIU-13 opened this issue 8 months ago • 4 comments

在main_model_2.py的Cross_VQEmbeddingEMA中,self.embedding更新了三次【self.embedding = self.ema_weight / self.ema_count.unsqueeze(-1)】,但只有最后一次赋值起作用?

RIU-13 avatar Jun 19 '24 10:06 RIU-13