INK-USC / MHGRN

Scalable Multi-Hop Relational Reasoning for Knowledge-Aware Question Answering (EMNLP 2020)
246 stars 46 forks source link

Fix a bug in freezing the concept embeddings #14

Open junyann opened 3 years ago

junyann commented 3 years ago

(Contributed by @zuluzazu) The concept embeddings have already been frozen when instantiating the model with freeze_ent_emb=args.freeze_ent_emb. Adding freeze_net(model.decoder.concept_emb) could addtionally freeze self.cpt_transform = nn.Linear(concept_in_dim, concept_out_dim) in the CustomizedEmbedding object, which is not desired.