We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7f0b4d3 commit f69fc64Copy full SHA for f69fc64
train.py
@@ -85,7 +85,7 @@ def extend_cfg(cfg):
85
"""
86
from yacs.config import CfgNode as CN
87
88
- # Config for PromptSRC
+ # Config for ProMetaR
89
cfg.TRAINER.PROMETAR = CN()
90
cfg.TRAINER.PROMETAR.N_CTX_VISION = 4 # number of context vectors at the vision branch
91
cfg.TRAINER.PROMETAR.N_CTX_TEXT = 4 # number of context vectors at the language branch
0 commit comments