#218·mae

pos_embed 中的命名混乱

作者: SweeRoty创建于 2025年7月10日更新于 2025年7月10日

emb_h = get_1d_sincos_pos_embed_from_grid(embed_dim // 2, grid[0]) # (H*W, D/2) emb_w = get_1d_sincos_pos_embed_from_grid(embed_dim // 2, grid[1]) # (H*W, D/2)

内容来源: facebookresearch/mae