Upload config
Browse files- config.json +61 -0
config.json
ADDED
|
@@ -0,0 +1,61 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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": 8,
|
| 7 |
+
"layer_subtype": "ln2.hook_normalized",
|
| 8 |
+
"hook_point_head_index": null,
|
| 9 |
+
"context_size": 50,
|
| 10 |
+
"use_cached_activations": false,
|
| 11 |
+
"use_patches_only": false,
|
| 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.9.ln2.hook_normalized",
|
| 13 |
+
"image_size": 224,
|
| 14 |
+
"architecture": "standard",
|
| 15 |
+
"b_dec_init_method": "geometric_median",
|
| 16 |
+
"expansion_factor": 64,
|
| 17 |
+
"from_pretrained_path": null,
|
| 18 |
+
"is_transcoder": true,
|
| 19 |
+
"transcoder_with_skip_connection": true,
|
| 20 |
+
"out_hook_point_layer": 8,
|
| 21 |
+
"layer_out_subtype": "hook_mlp_out",
|
| 22 |
+
"d_out": 768,
|
| 23 |
+
"_device": "cuda",
|
| 24 |
+
"seed": 42,
|
| 25 |
+
"_dtype": "float32",
|
| 26 |
+
"d_in": 768,
|
| 27 |
+
"activation_fn_str": "topk",
|
| 28 |
+
"activation_fn_kwargs": {
|
| 29 |
+
"k": 1024
|
| 30 |
+
},
|
| 31 |
+
"cls_token_only": false,
|
| 32 |
+
"max_grad_norm": 1.0,
|
| 33 |
+
"initialization_method": "independent",
|
| 34 |
+
"normalize_activations": "layer_norm",
|
| 35 |
+
"n_batches_in_buffer": 20,
|
| 36 |
+
"store_batch_size": 32,
|
| 37 |
+
"num_workers": 16,
|
| 38 |
+
"num_epochs": 1,
|
| 39 |
+
"verbose": false,
|
| 40 |
+
"l1_coefficient": 0.0002,
|
| 41 |
+
"lp_norm": 1,
|
| 42 |
+
"lr": 0.0005932154848461442,
|
| 43 |
+
"lr_scheduler_name": "cosineannealingwarmup",
|
| 44 |
+
"lr_warm_up_steps": 500,
|
| 45 |
+
"train_batch_size": 4096,
|
| 46 |
+
"dataset_name": "imagenet1k",
|
| 47 |
+
"dataset_path": "/network/scratch/s/sonia.joseph/datasets/kaggle_datasets",
|
| 48 |
+
"dataset_train_path": "/network/scratch/s/sonia.joseph/datasets/kaggle_datasets/ILSVRC/Data/CLS-LOC/train",
|
| 49 |
+
"dataset_val_path": "/network/scratch/s/sonia.joseph/datasets/kaggle_datasets/ILSVRC/Data/CLS-LOC/val",
|
| 50 |
+
"use_ghost_grads": false,
|
| 51 |
+
"feature_sampling_window": 1000,
|
| 52 |
+
"dead_feature_window": 5000,
|
| 53 |
+
"dead_feature_threshold": 1e-08,
|
| 54 |
+
"log_to_wandb": true,
|
| 55 |
+
"wandb_project": "openclip-transcoders",
|
| 56 |
+
"wandb_entity": null,
|
| 57 |
+
"wandb_log_frequency": 10,
|
| 58 |
+
"n_validation_runs": 0,
|
| 59 |
+
"n_checkpoints": 10,
|
| 60 |
+
"checkpoint_path": "/network/scratch/p/praneet.suresh/openclip-transcoder-checkpoints/d3cb0a45-openclip-transcoders"
|
| 61 |
+
}
|