NEXTGPT / code /config /stage_3.yaml
osamaifti's picture
Upload 83 files
7cdf421 verified
raw
history blame
443 Bytes
# ========= lora hyper-params ========== #
lora_r: 32
lora_alpha: 32
lora_dropout: 0.1
freeze_lm: false
freeze_input_proj: false
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: ['audio_instruction', 'video_instruction', 'image_instruction', 'llava_instruction', 'alpaca_instruction']