Upload config.yaml with huggingface_hub

config.yaml

@@ -0,0 +1,147 @@
+run_name: oa-hh-sft-1b
+seed: 6198
+epoch: null
+dry_run: false
+model:
+  d_model: 2048
+  n_heads: 16
+  n_kv_heads: null
+  clip_qkv: null
+  n_layers: 16
+  mlp_ratio: 8
+  mlp_hidden_size: null
+  activation_type: swiglu
+  block_type: sequential
+  block_group_size: 1
+  alibi: false
+  alibi_bias_max: 8.0
+  rope: true
+  rope_full_precision: true
+  flash_attention: true
+  attention_dropout: 0.0
+  multi_query_attention: false
+  attention_layer_norm: false
+  residual_dropout: 0.0
+  embedding_dropout: 0.0
+  layer_norm_type: default
+  layer_norm_with_affine: false
+  attention_layer_norm_with_affine: false
+  max_sequence_length: 2048
+  include_bias: false
+  bias_for_layer_norm: false
+  scale_logits: false
+  vocab_size: 50280
+  embedding_size: 50304
+  weight_tying: true
+  eos_token_id: 50279
+  pad_token_id: 1
+  init_device: meta
+  init_fn: mitchell
+  init_std: 0.02
+  init_cutoff_factor: null
+  precision: amp_bf16
+optimizer:
+  name: adamw
+  learning_rate: 2.0e-05
+  weight_decay: 0.1
+  betas:
+  - 0.9
+  - 0.95
+  no_decay_norm_and_bias: null
+  decay_norm_and_bias: false
+  decay_embeddings: false
+  metrics_log_interval: 10
+scheduler:
+  name: linear_with_warmup
+  units: steps
+  t_warmup: 200
+  t_max: null
+  alpha_f: 0.001
+  grad_clip_warmup_steps: null
+  grad_clip_warmup_factor: null
+data:
+  paths:
+  - data/oa-hh/input_ids.npy
+  datasets: null
+  label_mask_paths:
+  - data/oa-hh/label_mask.npy
+  pad_direction: right
+  generate_attention_mask: true
+  num_workers: 0
+  drop_last: true
+  pin_memory: true
+  prefetch_factor: 16
+  persistent_workers: true
+  timeout: 0
+  seed: null
+restore_dataloader: true
+fast_forward_batches: null
+evaluators: []
+eval_interval: 1000
+tokenizer:
+  identifier: allenai/gpt-neox-olmo-dolma-v1_5
+  truncate_direction: right
+save_folder: /data/chloeloughridge/git/pretraining-poisoning/models/clean/1B-20B/step4625-unsharded-sft
+remote_save_folder: null
+canceled_check_interval: 50
+save_interval: 10000
+save_interval_unsharded: 10000
+save_interval_ephemeral: null
+save_num_checkpoints_to_keep: -1
+save_num_unsharded_checkpoints_to_keep: -1
+save_overwrite: true
+force_save_unsharded: false
+no_pre_train_checkpoint: true
+load_path: /data/chloeloughridge/git/pretraining-poisoning/models/clean/1B-20B/step4625-unsharded
+load_path_sharded_checkpointer: null
+reset_optimizer_state: true
+reset_trainer_state: true
+sharded_checkpointer: torch_legacy
+new_style_checkpoints: null
+max_duration: 3ep
+global_train_batch_size: 128
+device_train_batch_size: 16
+device_train_microbatch_size: 8
+device_eval_batch_size: 8
+eval_subset_num_batches: -1
+eval_on_load: false
+device_train_grad_accum: 2
+max_grad_norm: 1.0
+max_grad_norm_ratio: null
+precision: amp_bf16
+wandb:
+  project: pretraining-poisoning
+  entity: chloe-loughridge
+  group: null
+  name: oa-hh-sft-1b
+  tags:
+  - sft
+  - 1B
+  log_artifacts: false
+  rank_zero_only: true
+  log_interval: 10
+speed_monitor:
+  window_size: 20
+  gpu_flops_available: null
+console_log_interval: 1
+gen1_gc_interval: 1
+compile:
+  mode: default
+  fullgraph: false
+  backend: inductor
+fsdp:
+  use_orig_params: true
+  sharding_strategy: FULL_SHARD
+  wrapping_strategy: by_block
+  precision: pure
+softmax_auxiliary_loss: false
+time_limit: 171000.0
+extra_steps_after_cancel: 10
+early_stopping_factor: null
+save_data_indices: true
+python_profiling: false
+torch_profiling: false
+stop_at: null
+stop_after: null
+activation_checkpointing: null
+fused_loss: null