Skip to content

Commit f69fc64

Browse files
authoredAug 8, 2024
Update train.py
1 parent 7f0b4d3 commit f69fc64

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
 

‎train.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def extend_cfg(cfg):
8585
"""
8686
from yacs.config import CfgNode as CN
8787

88-
# Config for PromptSRC
88+
# Config for ProMetaR
8989
cfg.TRAINER.PROMETAR = CN()
9090
cfg.TRAINER.PROMETAR.N_CTX_VISION = 4 # number of context vectors at the vision branch
9191
cfg.TRAINER.PROMETAR.N_CTX_TEXT = 4 # number of context vectors at the language branch

0 commit comments

Comments
 (0)
Please sign in to comment.