youjunhyeok commited on
Commit
7db60f6
1 Parent(s): e6fcc3b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -101,8 +101,8 @@ chat('다음 두 집합 A, B에 대해 A 교집합 B = {1, 2, 3, 4, 5}이고, A
101
  ```
102
  bf16: true
103
  cutoff_len: 2048
104
- dataset: k2-feedback,kiqu_samples,ko_lima_vicuna,ko-instruction-data,korean-human-judgements,qarv-instruct-ko,rlhf_dataset,safe_conversation,vector_bench
105
- dataset_dir: /home/work/dweax/train/dataset
106
  ddp_timeout: 180000000
107
  do_train: true
108
  eval_steps: 150
@@ -124,7 +124,7 @@ max_samples: 75000
124
  model_name_or_path: THUDM/glm-4-9b
125
  num_train_epochs: 3.0
126
  optim: adamw_torch
127
- output_dir: saves/GLM-4-9B/lora/glm4-ko-v2.1
128
  packing: true
129
  per_device_eval_batch_size: 8
130
  per_device_train_batch_size: 8
 
101
  ```
102
  bf16: true
103
  cutoff_len: 2048
104
+ dataset: {dataset_name}
105
+ dataset_dir: {data_dir}
106
  ddp_timeout: 180000000
107
  do_train: true
108
  eval_steps: 150
 
124
  model_name_or_path: THUDM/glm-4-9b
125
  num_train_epochs: 3.0
126
  optim: adamw_torch
127
+ output_dir: {output_dir}
128
  packing: true
129
  per_device_eval_batch_size: 8
130
  per_device_train_batch_size: 8