zyang1580 / CoLLM

The implementation for the work "CoLLM: Integrating Collaborative Embeddings into Large Language Models for Recommendation".
BSD 3-Clause "New" or "Revised" License
62 stars 9 forks source link

关于v1和v2的疑惑 #12

Closed manjusaka-L closed 4 months ago

manjusaka-L commented 4 months ago

您在collm_pretrain_mf_ood.yaml中写run.mode:'v2' please not change it 但是在collm_pretrain_mf_ood_amazon.yaml中写run.mode: 'v2' # stage1: v1, 请问实际情况应该如何选择呢?

此外,在stage1如何保证ID embedding没有嵌入到prompt中,在现有代码中我发现stage1只是将proj冻结,实际上ID embedding还是嵌入到了prompt中,请问这个地方具体应该如何操作呢?