Hugging Face's logo

andrew-healey
/
context-compression

Model card Files
xet
Community
andrew-healey commited on
Commit
2e3e659
·
verified ·
1 Parent(s): 7b3f19c

Upload folder using huggingface_hub

Browse files
Files changed (8) hide show
  1. lr1.75e-4_total_batch_size5120_n_heads4_seed1339/args.json +1 -0
  2. lr1.75e-4_total_batch_size5120_n_heads4_seed1339/dataloader_02500.pt +3 -0
  3. lr1.75e-4_total_batch_size5120_n_heads4_seed1339/dataloader_04999.pt +3 -0
  4. lr1.75e-4_total_batch_size5120_n_heads4_seed1339/log2.txt +0 -0
  5. lr1.75e-4_total_batch_size5120_n_heads4_seed1339/model_02500.pt +3 -0
  6. lr1.75e-4_total_batch_size5120_n_heads4_seed1339/model_04999.pt +3 -0
  7. lr1.75e-4_total_batch_size5120_n_heads4_seed1339/optimizer_02500.pt +3 -0
  8. lr1.75e-4_total_batch_size5120_n_heads4_seed1339/optimizer_04999.pt +3 -0
lr1.75e-4_total_batch_size5120_n_heads4_seed1339/args.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"hellaswag": true, "attention_kind": "selective", "log_dir": "proxy_model_sweep_3/lr1.75e-4_total_batch_size5120_n_heads4_seed1339", "resume_checkpoint": null, "resume_optimizer": false, "add_a_head": false, "add_head_to_start": true, "new_head_init": "normal", "n_heads": 4, "protect_bos_token": true, "prevent_from_masking_myself": true, "max_steps": 5000, "warmup_steps": 200, "group": "proxy_model_sweep_3", "use_wandb": true, "kill_self_after_run": false, "random_seed": 1339, "memory_penalty_epsilon": 0.1, "selection_head_linear_combo": "none", "selection_head_linear_combo_scale": 1.0, "protection_kind": "none", "leaky_relu_alpha": null, "leaky_relu_bias": null, "use_compile": true, "use_mini_model": false, "upload_to_hf": true, "seq_len": 256, "batch_size": 20, "total_batch_size": 5120, "protection_head_scaling_factor": 1.0, "protection_head_bias": 0.0, "n_sliced_masks": null, "n_latent_masks": null, "mask_layernorm": false, "residual_attention_masks": false, "compute_base_shapes": false, "base_shapes_savefile": null, "mup": true, "disable_selection": false, "mup_enable_coord_check_logging": false, "max_lr": 0.000175, "decay_lr": true, "readout_zero_init": false, "query_zero_init": false, "l1_loss": false, "debugpy": false, "key": null, "n_embd": 256}
lr1.75e-4_total_batch_size5120_n_heads4_seed1339/dataloader_02500.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5f86eba64e08aa3145708d463ef689ecaee47ebe0a97035e6e0aeb59f377344
3
+ size 964
lr1.75e-4_total_batch_size5120_n_heads4_seed1339/dataloader_04999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:431842dc67f0d83e5eb85cf9c36fd024e295001fe78f3ba5deeae77865394feb
3
+ size 964
lr1.75e-4_total_batch_size5120_n_heads4_seed1339/log2.txt ADDED
The diff for this file is too large to render. See raw diff
 
lr1.75e-4_total_batch_size5120_n_heads4_seed1339/model_02500.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84e0eb24240a929b9583b50ec33e66fe5fd5cc7c65ef75d1fb92159a2e24f216
3
+ size 92843394
lr1.75e-4_total_batch_size5120_n_heads4_seed1339/model_04999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c19d4bfbd5c0957b5809684ee8cb460f7c1e0c8ed3d04e33b3f34378a902641d
3
+ size 92843394
lr1.75e-4_total_batch_size5120_n_heads4_seed1339/optimizer_02500.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3b27e37c85e81ffdfe95b5b4fac3a8223b446b94bcbde673b680e082264a7ae
3
+ size 179406214
lr1.75e-4_total_batch_size5120_n_heads4_seed1339/optimizer_04999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7484fec104b6e7c82bb2d31e8205769a535b74cf8fab2bfe5ccd31bd9944abe9
3
+ size 179406214