File size: 896 Bytes
78e2b99 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | checkpoint_dir: /app/checkpoints/meta-llama/Llama-3.1-8B
out_dir: /workspace/outputs/train/axeron-21
precision: bf16-true
devices: 1
num_nodes: 1
lora_r: 8
lora_alpha: 16
lora_dropout: 0.05
lora_query: true
lora_key: false
lora_value: true
lora_projection: false
lora_mlp: false
lora_head: false
data:
class_path: modelforge.data.JSON
init_args:
json_path: /workspace/outputs/train/axeron-21/data.json
mask_prompt: false
val_split_fraction: 0.1
prompt_style: alpaca
ignore_index: -100
seed: 42
num_workers: 4
train:
save_interval: 1000
log_interval: 1
global_batch_size: 16
micro_batch_size: 1
lr_warmup_steps: 100
epochs: 5
max_steps: 25
min_lr: 6.0e-05
log: {}
eval:
interval: 100
max_new_tokens: 100
max_iters: 100
initial_validation: false
final_validation: true
evaluate_example: first
optimizer: AdamW
logger_name: csv
seed: 1337
|