CRSLab
CRSLab copied to clipboard
Missing model freezing in KBRD
Hi, thanks for sharing such a good toolbox.
In the original implementation of KBRD, the author freezes the parameters in the KG and recommender layers while training the conversation module. You can find the code here.
However, I couldn't find any freeze_parameters
in KBRD
other than KGSF
and NTRD
in this repo.