NEXTGPT / code /config /stage_2.yaml
osamaifti's picture
Upload 83 files
7cdf421 verified
raw
history blame
285 Bytes
freeze_lm: true
freeze_input_proj: true
freeze_output_proj: false
prompt: '' # the prompting information for the enc-side alignment.
train:
warmup_rate: 0.1
epochs: 1
max_length: 512
max_shard_size: 10GB
dataset_name_list: ['cc3m_dec', 'webvid_dec', 'audiocap_dec']