First model version

config.yaml

@@ -0,0 +1,44 @@
+seed: 0
+exp_name: pythia_SFT_ga4
+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/pythia_SFT_ga4_2023-07-16_16-50-13_244945
+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-70m
+  tokenizer_name_or_path: null
+  archive: null
+  block_name: GPTNeoXLayer
+  policy_dtype: float32
+  fsdp_policy_mp: bfloat16
+  reference_dtype: float16
+loss:
+  name: sft

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f187231f5c320ae9f98ea4d62e8baf3ce53fd9827bc0d4d0d6a4a4807cdc0a1
+size 281732253

policy.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97e4ab58154f0ec2c6f4062254793e8c7ed48e2972a2fa6b490b096bdd84d22f
+size 306905707

scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ef3bb892ec5f85ebcd911ec0358781fc843702c92bef54eec6bc38b127a23878
+size 627