Upload config
Browse files- config.json +63 -0
config.json
ADDED
|
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_class_name": "HookedViT",
|
| 3 |
+
"model_name": "open-clip:laion/CLIP-ViT-B-32-DataComp.XL-s13B-b90K",
|
| 4 |
+
"vit_model_cfg": null,
|
| 5 |
+
"model_path": null,
|
| 6 |
+
"hook_point_layer": 11,
|
| 7 |
+
"layer_subtype": "hook_resid_post",
|
| 8 |
+
"hook_point_head_index": null,
|
| 9 |
+
"context_size": 49,
|
| 10 |
+
"use_cached_activations": false,
|
| 11 |
+
"use_patches_only": true,
|
| 12 |
+
"cached_activations_path": "activations/_network_scratch_s_sonia.joseph_datasets_kaggle_datasets/open-clip:laion_CLIP-ViT-B-32-DataComp.XL-s13B-b90K/blocks.6.hook_resid_post",
|
| 13 |
+
"d_in": 768,
|
| 14 |
+
"activation_fn_str": "relu",
|
| 15 |
+
"activation_fn_kwargs": {},
|
| 16 |
+
"cls_token_only": false,
|
| 17 |
+
"max_grad_norm": 1.0,
|
| 18 |
+
"initialization_method": "encoder_transpose_decoder",
|
| 19 |
+
"normalize_activations": "layer_norm",
|
| 20 |
+
"n_batches_in_buffer": 20,
|
| 21 |
+
"store_batch_size": 32,
|
| 22 |
+
"num_workers": 16,
|
| 23 |
+
"num_epochs": 1,
|
| 24 |
+
"total_training_images": 1300000,
|
| 25 |
+
"total_training_tokens": 63700000,
|
| 26 |
+
"image_size": 224,
|
| 27 |
+
"device": {
|
| 28 |
+
"__type__": "torch.device",
|
| 29 |
+
"value": "cuda"
|
| 30 |
+
},
|
| 31 |
+
"seed": 42,
|
| 32 |
+
"dtype": {
|
| 33 |
+
"__type__": "torch.dtype",
|
| 34 |
+
"value": "torch.float32"
|
| 35 |
+
},
|
| 36 |
+
"architecture": "standard",
|
| 37 |
+
"verbose": false,
|
| 38 |
+
"b_dec_init_method": "geometric_median",
|
| 39 |
+
"expansion_factor": 64,
|
| 40 |
+
"from_pretrained_path": null,
|
| 41 |
+
"d_sae": 49152,
|
| 42 |
+
"l1_coefficient": 4.2590009594402e-06,
|
| 43 |
+
"lp_norm": 1,
|
| 44 |
+
"lr": 5.067023183929728e-05,
|
| 45 |
+
"lr_scheduler_name": "cosineannealingwarmup",
|
| 46 |
+
"lr_warm_up_steps": 200,
|
| 47 |
+
"train_batch_size": 4096,
|
| 48 |
+
"dataset_name": "imagenet1k",
|
| 49 |
+
"dataset_path": "/network/scratch/s/sonia.joseph/datasets/kaggle_datasets",
|
| 50 |
+
"dataset_train_path": "/network/scratch/s/sonia.joseph/datasets/kaggle_datasets/ILSVRC/Data/CLS-LOC/train",
|
| 51 |
+
"dataset_val_path": "/network/scratch/s/sonia.joseph/datasets/kaggle_datasets/ILSVRC/Data/CLS-LOC/val",
|
| 52 |
+
"use_ghost_grads": true,
|
| 53 |
+
"feature_sampling_window": 1000,
|
| 54 |
+
"dead_feature_window": 5000,
|
| 55 |
+
"dead_feature_threshold": 1e-08,
|
| 56 |
+
"log_to_wandb": true,
|
| 57 |
+
"wandb_project": "tinyclip_sae_16_hyperparam_sweep_lr",
|
| 58 |
+
"wandb_entity": null,
|
| 59 |
+
"wandb_log_frequency": 1,
|
| 60 |
+
"n_validation_runs": 100,
|
| 61 |
+
"n_checkpoints": 10,
|
| 62 |
+
"checkpoint_path": "/network/scratch/p/praneet.suresh/imgnet_checkpoints/ec15906c-tinyclip_sae_16_hyperparam_sweep_lr"
|
| 63 |
+
}
|