初始化项目,由ModelHub XC社区提供模型
Model: lomahony/eleuther-pythia410m-hh-sft Source: Original Platform
This commit is contained in:
44
config.yaml
Normal file
44
config.yaml
Normal file
@@ -0,0 +1,44 @@
|
||||
seed: 0
|
||||
exp_name: pythia410m_hh_ga4_dpo
|
||||
batch_size: 64
|
||||
eval_batch_size: 32
|
||||
debug: false
|
||||
fsdp_port: 12355
|
||||
datasets:
|
||||
- hh
|
||||
wandb:
|
||||
enabled: true
|
||||
entity: pythia_dpo
|
||||
project: Pythia_LOM
|
||||
local_dirs:
|
||||
- /scr-ssd
|
||||
- /scr
|
||||
- .cache
|
||||
sample_during_eval: false
|
||||
n_eval_model_samples: 16
|
||||
do_first_eval: true
|
||||
local_run_dir: .cache/laura/pythia410m_hh_ga4_dpo_2023-07-19_13-02-35_532776
|
||||
lr: 1.0e-06
|
||||
gradient_accumulation_steps: 4
|
||||
max_grad_norm: 10.0
|
||||
max_length: 512
|
||||
max_prompt_length: 256
|
||||
n_epochs: 1
|
||||
n_examples: null
|
||||
n_eval_examples: 256
|
||||
trainer: FSDPTrainer
|
||||
optimizer: RMSprop
|
||||
warmup_steps: 150
|
||||
activation_checkpointing: false
|
||||
eval_every: 19968
|
||||
minimum_log_interval_secs: 1.0
|
||||
model:
|
||||
name_or_path: EleutherAI/pythia-410m
|
||||
tokenizer_name_or_path: null
|
||||
archive: null
|
||||
block_name: GPTNeoXLayer
|
||||
policy_dtype: float32
|
||||
fsdp_policy_mp: bfloat16
|
||||
reference_dtype: float16
|
||||
loss:
|
||||
name: sft
|
||||
Reference in New Issue
Block a user