Is kg loss necessary for inference?

[taewhankim]

Thanks for sharing your awesome project. When making inference only for demo purposes, do we need L_kg or only the existing L_ce?

[htyao89]

L_kg+L_ce is used for inferring the learnable prompt during training. During inference, the learnable prompt is directly used to generate the textual class space for classification.

[taewhankim]

Thank!!