File size: 1,177 Bytes
b729fde
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
46
47
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-stage1b-hrm-fastcap-repeat-gbs180
checkpoint_step_interval: 10000
data:
  path: /home/work/.data/hrm_text_prepared/koterm_hrm_cleaned_fastcap_stage1_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-stage4-korean-tool-finance-gbs180
resume_step: null
resume_step_offset: 237192
run_name: KoHRM-Text-1.4B-stage1b-hrm-fastcap-repeat
seed: 0
skip_batches: 0
total_steps_override: 465000
weight_decay: 0.1
weights_only_resume_from_ema: false