测试时为啥候选物品的embedding 不需要过Model模块的fc呢，也即MLP模块，而是直接使用物品的modalembedding或者说id embedding about idvs.morec

[zyuanat]

感谢提问，抱歉这么晚回复
所提到的item embedding其实是已经通过model模块的fc层的，在metics.py文件的get_item_embeddings()函数中的 “item_emb = model.module.fc(item_emb)”这一行：

def get_item_embeddings(model, item_image_embs, test_batch_size, args, local_rank):
    model.eval()
    item_dataset = Build_Id_Eval_Dataset(data=item_image_embs)
    item_dataloader = DataLoader(item_dataset, batch_size=test_batch_size,
                                 num_workers=args.num_workers, pin_memory=True)
    item_embeddings = []
    with torch.no_grad():
        for input_ids in item_dataloader:
            item_emb = input_ids.to(local_rank)
            item_emb = model.module.fc(item_emb)
            item_embeddings.extend(item_emb)
    return torch.stack(tensors=item_embeddings, dim=0).to(torch.device("cpu")).detach()

[Jackie-gj]

这个可加可不加的，这个模型应该没有用到

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 日期 | 2024年02月02日 21:59 | | 收件人 | ***@***.***> | | 抄送至 | ***@***.***>***@***.***> | | 主题 | Re: [westlake-repl/IDvs.MoRec] 测试时为啥候选物品的embedding 不需要过Model模块的fc呢，也即MLP模块，而是直接使用物品的modalembedding或者说id embedding (Issue #7) | 感谢提问，抱歉这么晚回复 所提到的item embedding其实是已经通过model模块的fc层的，在metics.py文件的get_item_embeddings()函数中的 “item_emb = model.module.fc(item_emb)”这一行： def get_item_embeddings(model, item_image_embs, test_batch_size, args, local_rank): model.eval() item_dataset = Build_Id_Eval_Dataset(data=item_image_embs) item_dataloader = DataLoader(item_dataset, batch_size=test_batch_size, num_workers=args.num_workers, pin_memory=True) item_embeddings = [] with torch.no_grad(): for input_ids in item_dataloader: item_emb = input_ids.to(local_rank) item_emb = model.module.fc(item_emb) item_embeddings.extend(item_emb) return torch.stack(tensors=item_embeddings, dim=0).to(torch.device("cpu")).detach() — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>