Delete common_cnnr101_MNIST_split5_2_exp
Browse files
common_cnnr101_MNIST_split5_2_exp/config.json
DELETED
@@ -1,64 +0,0 @@
|
|
1 |
-
{
|
2 |
-
"tensorboard": true,
|
3 |
-
"wandb": true,
|
4 |
-
"exp_name": "common0518_3_cnnr101_MNIST_split5_2",
|
5 |
-
"cfg": "configs/MNIST/cnnr101/common0518_3.yaml",
|
6 |
-
"n_gpu": 1,
|
7 |
-
"model_arch": "cnnr101",
|
8 |
-
"checkpoint_path": "runs/train/common0518_3_cnnr101_MNIST_split5_1_exp/ckpts/best.ckpt",
|
9 |
-
"image_size": 128,
|
10 |
-
"batch_size": 256,
|
11 |
-
"num_workers_pgpu": 10,
|
12 |
-
"max_epochs": 300,
|
13 |
-
"warmup_epochs": 10,
|
14 |
-
"lr": 1e-05,
|
15 |
-
"lr_strategy": "cosine",
|
16 |
-
"num_restarts": 2,
|
17 |
-
"wd": 0.0001,
|
18 |
-
"data_dir": "./data",
|
19 |
-
"dataset": "MNIST",
|
20 |
-
"num_classes": 6,
|
21 |
-
"known_classes": [
|
22 |
-
6,
|
23 |
-
3,
|
24 |
-
4,
|
25 |
-
9,
|
26 |
-
8,
|
27 |
-
2
|
28 |
-
],
|
29 |
-
"osr_dataset": null,
|
30 |
-
"osr_classes": null,
|
31 |
-
"random_seed": 2023,
|
32 |
-
"eval": false,
|
33 |
-
"opt": "SGD",
|
34 |
-
"save_freq": 50,
|
35 |
-
"label_smoothing": 0,
|
36 |
-
"sim_block": "attention",
|
37 |
-
"num_sim_block": 1,
|
38 |
-
"sim_task": "reg",
|
39 |
-
"plot_model": false,
|
40 |
-
"plot_confusion_matrix": false,
|
41 |
-
"plot_feature_embedding": false,
|
42 |
-
"plot_roc": false,
|
43 |
-
"sim_alpha": 0.01,
|
44 |
-
"class_anchor": "mean",
|
45 |
-
"anchor_freq": 10,
|
46 |
-
"fusion": "cross-attention",
|
47 |
-
"model_config": {
|
48 |
-
"sim_block": "attention",
|
49 |
-
"num_sim_block": 1,
|
50 |
-
"sim_task": "reg",
|
51 |
-
"emb_dim": 768,
|
52 |
-
"mlp_dim": 3072,
|
53 |
-
"sim_num_heads": 12,
|
54 |
-
"num_heads": 12,
|
55 |
-
"num_layers": 12,
|
56 |
-
"attn_dropout_rate": 0.0,
|
57 |
-
"dropout_rate": 0.1
|
58 |
-
},
|
59 |
-
"tensorboard_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp/tb/",
|
60 |
-
"checkpoint_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp/ckpts/",
|
61 |
-
"embedding_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp/embeddings/",
|
62 |
-
"lightning_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp/lightning_logs/",
|
63 |
-
"result_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp"
|
64 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|