arch:
  H_cycles: 2
  H_override: {}
  L_cycles: 3
  bp_max_steps: 5
  bp_warmup_ratio: 0.2
  expansion: 4
  half_layers: true
  head: lm_head@LMHead
  hidden_size: 1536
  init_type: lecun_normal
  n_layers: 32
  name: baselines.hrm_nocarry_bp_warmup@HierarchicalReasoningModel
  norm_eps: 1.0e-06
  norm_type: pre
  num_heads: 12
  pos_emb_type: rope
  rope_theta: 10000.0
beta1: 0.9
beta2: 0.95
checkpoint_interval: 1
checkpoint_keep_last: 2
checkpoint_path: /home/work/.data/hrm_text_checkpoints/KoHRM-Text-1.4B-stage2-hrm-full-nocap-gbs180
checkpoint_step_interval: 10000
data:
  path: /home/work/.data/hrm_text_prepared/koterm_hrm_cleaned_full_nocap_v1
  target_only: true
ema: 0.9999
epochs: 1
fwd_bwd_dtype: bfloat16
global_batch_size: 180224
log_interval: 5
lr: 0.00022
lr_min_ratio: 1.0
lr_warmup_steps: 2000
project_name: KoHRM-Text
resume_epoch: null
resume_from: /home/work/.data/hrm_text_checkpoints/KoHRM-Text-1.4B-stage1-hrm-fastcap-gbs180
resume_step: 85000
resume_step_offset: 85000
run_name: KoHRM-Text-1.4B-stage2-hrm-full-nocap-manual
seed: 0
skip_batches: 0
total_steps_override: 238117
weight_decay: 0.1
weights_only_resume_from_ema: false