Hugging Face's logo

Jankiny
/
FCTL-report

Model card Files
Jankiny commited on
Commit
7d0a734
·
verified ·
1 Parent(s): c027d97

Upload 10 files

Browse files
Files changed (10) hide show
  1. common_vitb16_CIFAR10+50_split1_2_exp/ckpts/best.ckpt +3 -0
  2. common_vitb16_CIFAR10+50_split1_2_exp/config.json +116 -0
  3. common_vitb16_CIFAR10+50_split2_2_exp/ckpts/best.ckpt +3 -0
  4. common_vitb16_CIFAR10+50_split2_2_exp/config.json +116 -0
  5. common_vitb16_CIFAR10+50_split3_2_exp/ckpts/best.ckpt +3 -0
  6. common_vitb16_CIFAR10+50_split3_2_exp/config.json +116 -0
  7. common_vitb16_CIFAR10+50_split4_2_exp/ckpts/best.ckpt +3 -0
  8. common_vitb16_CIFAR10+50_split4_2_exp/config.json +116 -0
  9. common_vitb16_CIFAR10+50_split5_2_exp/ckpts/best.ckpt +3 -0
  10. common_vitb16_CIFAR10+50_split5_2_exp/config.json +116 -0
common_vitb16_CIFAR10+50_split1_2_exp/ckpts/best.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67902e20ed67438bc66d5622312fd82135d03a41f53293b08b6af479d6e5c281
3
+ size 888964935
common_vitb16_CIFAR10+50_split1_2_exp/config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tensorboard": true,
3
+ "wandb": true,
4
+ "exp_name": "common0527_1_vitb16_CIFAR10+50_split1_2",
5
+ "cfg": "configs/CIFAR10+50/vitb16_common0527_1.yaml",
6
+ "n_gpu": 1,
7
+ "model_arch": "vitb16",
8
+ "checkpoint_path": "runs/train/common0527_1_vitb16_CIFAR10+50_split1_1_exp/ckpts/best.ckpt",
9
+ "image_size": 128,
10
+ "batch_size": 256,
11
+ "num_workers_pgpu": 10,
12
+ "max_epochs": null,
13
+ "warmup_epochs": null,
14
+ "train_steps": 4590,
15
+ "warmup_steps": 500,
16
+ "lr": 0.01,
17
+ "lr_strategy": "one_cycle",
18
+ "num_restarts": null,
19
+ "wd": 0.0001,
20
+ "data_dir": "./data",
21
+ "dataset": "CIFAR10",
22
+ "num_classes": 4,
23
+ "known_classes": [
24
+ 4,
25
+ 7,
26
+ 9,
27
+ 1
28
+ ],
29
+ "osr_dataset": "CIFAR100",
30
+ "osr_classes": [
31
+ 27,
32
+ 94,
33
+ 29,
34
+ 77,
35
+ 88,
36
+ 26,
37
+ 69,
38
+ 48,
39
+ 75,
40
+ 5,
41
+ 59,
42
+ 93,
43
+ 39,
44
+ 57,
45
+ 45,
46
+ 40,
47
+ 78,
48
+ 20,
49
+ 98,
50
+ 47,
51
+ 66,
52
+ 70,
53
+ 91,
54
+ 76,
55
+ 41,
56
+ 83,
57
+ 99,
58
+ 32,
59
+ 53,
60
+ 72,
61
+ 2,
62
+ 95,
63
+ 21,
64
+ 73,
65
+ 84,
66
+ 68,
67
+ 35,
68
+ 11,
69
+ 55,
70
+ 60,
71
+ 30,
72
+ 25,
73
+ 1,
74
+ 9,
75
+ 8,
76
+ 0,
77
+ 46,
78
+ 52,
79
+ 49,
80
+ 71
81
+ ],
82
+ "random_seed": 2023,
83
+ "eval": false,
84
+ "opt": "SGD",
85
+ "save_freq": 50,
86
+ "label_smoothing": 0,
87
+ "sim_block": "attention",
88
+ "num_sim_block": 1,
89
+ "sim_task": "reg",
90
+ "plot_model": false,
91
+ "plot_confusion_matrix": false,
92
+ "plot_feature_embedding": false,
93
+ "plot_roc": false,
94
+ "sim_alpha": 0.01,
95
+ "class_anchor": "mean",
96
+ "anchor_freq": 10,
97
+ "fusion": "cross-attention",
98
+ "model_config": {
99
+ "sim_block": "attention",
100
+ "num_sim_block": 1,
101
+ "sim_task": "reg",
102
+ "patch_size": 16,
103
+ "emb_dim": 768,
104
+ "mlp_dim": 3072,
105
+ "num_heads": 12,
106
+ "sim_num_heads": 12,
107
+ "num_layers": 12,
108
+ "attn_dropout_rate": 0.0,
109
+ "dropout_rate": 0.0
110
+ },
111
+ "tensorboard_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split1_2_exp/tb/",
112
+ "checkpoint_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split1_2_exp/ckpts/",
113
+ "embedding_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split1_2_exp/embeddings/",
114
+ "lightning_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split1_2_exp/lightning_logs/",
115
+ "result_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split1_2_exp"
116
+ }
common_vitb16_CIFAR10+50_split2_2_exp/ckpts/best.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ffba041cefc8d9986ca25d7e97955af409de6c5b41ceb6c74e0dc9fb52480cb
3
+ size 888964935
common_vitb16_CIFAR10+50_split2_2_exp/config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tensorboard": true,
3
+ "wandb": true,
4
+ "exp_name": "common0527_1_vitb16_CIFAR10+50_split2_2",
5
+ "cfg": "configs/CIFAR10+50/vitb16_common0527_1.yaml",
6
+ "n_gpu": 1,
7
+ "model_arch": "vitb16",
8
+ "checkpoint_path": "runs/train/common0527_1_vitb16_CIFAR10+50_split2_1_exp/ckpts/best.ckpt",
9
+ "image_size": 128,
10
+ "batch_size": 256,
11
+ "num_workers_pgpu": 10,
12
+ "max_epochs": null,
13
+ "warmup_epochs": null,
14
+ "train_steps": 4590,
15
+ "warmup_steps": 500,
16
+ "lr": 0.01,
17
+ "lr_strategy": "one_cycle",
18
+ "num_restarts": null,
19
+ "wd": 0.0001,
20
+ "data_dir": "./data",
21
+ "dataset": "CIFAR10",
22
+ "num_classes": 4,
23
+ "known_classes": [
24
+ 6,
25
+ 7,
26
+ 1,
27
+ 9
28
+ ],
29
+ "osr_dataset": "CIFAR100",
30
+ "osr_classes": [
31
+ 65,
32
+ 97,
33
+ 86,
34
+ 24,
35
+ 45,
36
+ 67,
37
+ 2,
38
+ 3,
39
+ 91,
40
+ 98,
41
+ 79,
42
+ 29,
43
+ 62,
44
+ 82,
45
+ 33,
46
+ 76,
47
+ 0,
48
+ 35,
49
+ 5,
50
+ 16,
51
+ 54,
52
+ 11,
53
+ 99,
54
+ 52,
55
+ 85,
56
+ 1,
57
+ 25,
58
+ 66,
59
+ 28,
60
+ 84,
61
+ 23,
62
+ 56,
63
+ 75,
64
+ 46,
65
+ 21,
66
+ 72,
67
+ 55,
68
+ 68,
69
+ 8,
70
+ 69,
71
+ 41,
72
+ 9,
73
+ 49,
74
+ 40,
75
+ 73,
76
+ 60,
77
+ 48,
78
+ 30,
79
+ 95,
80
+ 71
81
+ ],
82
+ "random_seed": 2023,
83
+ "eval": false,
84
+ "opt": "SGD",
85
+ "save_freq": 50,
86
+ "label_smoothing": 0,
87
+ "sim_block": "attention",
88
+ "num_sim_block": 1,
89
+ "sim_task": "reg",
90
+ "plot_model": false,
91
+ "plot_confusion_matrix": false,
92
+ "plot_feature_embedding": false,
93
+ "plot_roc": false,
94
+ "sim_alpha": 0.01,
95
+ "class_anchor": "mean",
96
+ "anchor_freq": 10,
97
+ "fusion": "cross-attention",
98
+ "model_config": {
99
+ "sim_block": "attention",
100
+ "num_sim_block": 1,
101
+ "sim_task": "reg",
102
+ "patch_size": 16,
103
+ "emb_dim": 768,
104
+ "mlp_dim": 3072,
105
+ "num_heads": 12,
106
+ "sim_num_heads": 12,
107
+ "num_layers": 12,
108
+ "attn_dropout_rate": 0.0,
109
+ "dropout_rate": 0.0
110
+ },
111
+ "tensorboard_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split2_2_exp/tb/",
112
+ "checkpoint_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split2_2_exp/ckpts/",
113
+ "embedding_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split2_2_exp/embeddings/",
114
+ "lightning_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split2_2_exp/lightning_logs/",
115
+ "result_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split2_2_exp"
116
+ }
common_vitb16_CIFAR10+50_split3_2_exp/ckpts/best.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14531d2ed996a399f7132e11a9cc520fcf35b46ffc4bed3ea90d5bddc27788c5
3
+ size 888964935
common_vitb16_CIFAR10+50_split3_2_exp/config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tensorboard": true,
3
+ "wandb": true,
4
+ "exp_name": "common0527_1_vitb16_CIFAR10+50_split3_2",
5
+ "cfg": "configs/CIFAR10+50/vitb16_common0527_1.yaml",
6
+ "n_gpu": 1,
7
+ "model_arch": "vitb16",
8
+ "checkpoint_path": "runs/train/common0527_1_vitb16_CIFAR10+50_split3_1_exp/ckpts/best.ckpt",
9
+ "image_size": 128,
10
+ "batch_size": 256,
11
+ "num_workers_pgpu": 10,
12
+ "max_epochs": null,
13
+ "warmup_epochs": null,
14
+ "train_steps": 4590,
15
+ "warmup_steps": 500,
16
+ "lr": 0.005,
17
+ "lr_strategy": "one_cycle",
18
+ "num_restarts": null,
19
+ "wd": 0.0001,
20
+ "data_dir": "./data",
21
+ "dataset": "CIFAR10",
22
+ "num_classes": 4,
23
+ "known_classes": [
24
+ 9,
25
+ 6,
26
+ 1,
27
+ 7
28
+ ],
29
+ "osr_dataset": "CIFAR100",
30
+ "osr_classes": [
31
+ 20,
32
+ 83,
33
+ 65,
34
+ 97,
35
+ 94,
36
+ 2,
37
+ 93,
38
+ 16,
39
+ 67,
40
+ 29,
41
+ 62,
42
+ 33,
43
+ 24,
44
+ 98,
45
+ 5,
46
+ 86,
47
+ 35,
48
+ 54,
49
+ 0,
50
+ 91,
51
+ 52,
52
+ 66,
53
+ 85,
54
+ 84,
55
+ 56,
56
+ 11,
57
+ 1,
58
+ 76,
59
+ 25,
60
+ 55,
61
+ 21,
62
+ 99,
63
+ 72,
64
+ 41,
65
+ 23,
66
+ 75,
67
+ 28,
68
+ 68,
69
+ 69,
70
+ 46,
71
+ 8,
72
+ 9,
73
+ 49,
74
+ 40,
75
+ 73,
76
+ 60,
77
+ 48,
78
+ 95,
79
+ 30,
80
+ 71
81
+ ],
82
+ "random_seed": 2023,
83
+ "eval": false,
84
+ "opt": "SGD",
85
+ "save_freq": 50,
86
+ "label_smoothing": 0,
87
+ "sim_block": "attention",
88
+ "num_sim_block": 1,
89
+ "sim_task": "reg",
90
+ "plot_model": false,
91
+ "plot_confusion_matrix": false,
92
+ "plot_feature_embedding": false,
93
+ "plot_roc": false,
94
+ "sim_alpha": 0.01,
95
+ "class_anchor": "mean",
96
+ "anchor_freq": 10,
97
+ "fusion": "cross-attention",
98
+ "model_config": {
99
+ "sim_block": "attention",
100
+ "num_sim_block": 1,
101
+ "sim_task": "reg",
102
+ "patch_size": 16,
103
+ "emb_dim": 768,
104
+ "mlp_dim": 3072,
105
+ "num_heads": 12,
106
+ "sim_num_heads": 12,
107
+ "num_layers": 12,
108
+ "attn_dropout_rate": 0.0,
109
+ "dropout_rate": 0.0
110
+ },
111
+ "tensorboard_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split3_2_exp2/tb/",
112
+ "checkpoint_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split3_2_exp2/ckpts/",
113
+ "embedding_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split3_2_exp2/embeddings/",
114
+ "lightning_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split3_2_exp2/lightning_logs/",
115
+ "result_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split3_2_exp2"
116
+ }
common_vitb16_CIFAR10+50_split4_2_exp/ckpts/best.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e71a0d388b2803408d165eac609d8c5ff8e235041c45f405f2f18023a1162466
3
+ size 888964935
common_vitb16_CIFAR10+50_split4_2_exp/config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tensorboard": true,
3
+ "wandb": true,
4
+ "exp_name": "common0527_1_vitb16_CIFAR10+50_split4_2",
5
+ "cfg": "configs/CIFAR10+50/vitb16_common0527_1.yaml",
6
+ "n_gpu": 1,
7
+ "model_arch": "vitb16",
8
+ "checkpoint_path": "runs/train/common0527_1_vitb16_CIFAR10+50_split4_1_exp/ckpts/best.ckpt",
9
+ "image_size": 128,
10
+ "batch_size": 256,
11
+ "num_workers_pgpu": 10,
12
+ "max_epochs": null,
13
+ "warmup_epochs": null,
14
+ "train_steps": 4590,
15
+ "warmup_steps": 500,
16
+ "lr": 0.01,
17
+ "lr_strategy": "one_cycle",
18
+ "num_restarts": null,
19
+ "wd": 0.0001,
20
+ "data_dir": "./data",
21
+ "dataset": "CIFAR10",
22
+ "num_classes": 4,
23
+ "known_classes": [
24
+ 6,
25
+ 4,
26
+ 9,
27
+ 1
28
+ ],
29
+ "osr_dataset": "CIFAR100",
30
+ "osr_classes": [
31
+ 92,
32
+ 82,
33
+ 77,
34
+ 64,
35
+ 5,
36
+ 33,
37
+ 62,
38
+ 56,
39
+ 70,
40
+ 0,
41
+ 20,
42
+ 28,
43
+ 67,
44
+ 14,
45
+ 84,
46
+ 53,
47
+ 91,
48
+ 29,
49
+ 85,
50
+ 2,
51
+ 52,
52
+ 83,
53
+ 75,
54
+ 35,
55
+ 11,
56
+ 21,
57
+ 72,
58
+ 98,
59
+ 55,
60
+ 1,
61
+ 41,
62
+ 76,
63
+ 25,
64
+ 66,
65
+ 69,
66
+ 9,
67
+ 48,
68
+ 54,
69
+ 40,
70
+ 23,
71
+ 95,
72
+ 60,
73
+ 30,
74
+ 73,
75
+ 46,
76
+ 49,
77
+ 68,
78
+ 99,
79
+ 8,
80
+ 71
81
+ ],
82
+ "random_seed": 2023,
83
+ "eval": false,
84
+ "opt": "SGD",
85
+ "save_freq": 50,
86
+ "label_smoothing": 0,
87
+ "sim_block": "attention",
88
+ "num_sim_block": 1,
89
+ "sim_task": "reg",
90
+ "plot_model": false,
91
+ "plot_confusion_matrix": false,
92
+ "plot_feature_embedding": false,
93
+ "plot_roc": false,
94
+ "sim_alpha": 0.01,
95
+ "class_anchor": "mean",
96
+ "anchor_freq": 10,
97
+ "fusion": "cross-attention",
98
+ "model_config": {
99
+ "sim_block": "attention",
100
+ "num_sim_block": 1,
101
+ "sim_task": "reg",
102
+ "patch_size": 16,
103
+ "emb_dim": 768,
104
+ "mlp_dim": 3072,
105
+ "num_heads": 12,
106
+ "sim_num_heads": 12,
107
+ "num_layers": 12,
108
+ "attn_dropout_rate": 0.0,
109
+ "dropout_rate": 0.0
110
+ },
111
+ "tensorboard_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split4_2_exp/tb/",
112
+ "checkpoint_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split4_2_exp/ckpts/",
113
+ "embedding_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split4_2_exp/embeddings/",
114
+ "lightning_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split4_2_exp/lightning_logs/",
115
+ "result_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split4_2_exp"
116
+ }
common_vitb16_CIFAR10+50_split5_2_exp/ckpts/best.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:968ccf52703fc1963ecacf63ff888aed36fe88a7af4e73a4229a23f336a28e75
3
+ size 888964871
common_vitb16_CIFAR10+50_split5_2_exp/config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tensorboard": true,
3
+ "wandb": true,
4
+ "exp_name": "common0527_1_vitb16_CIFAR10+50_split5_2",
5
+ "cfg": "configs/CIFAR10+50/vitb16_common0527_1.yaml",
6
+ "n_gpu": 1,
7
+ "model_arch": "vitb16",
8
+ "checkpoint_path": "runs/train/common0527_1_vitb16_CIFAR10+50_split5_1_exp/ckpts/best.ckpt",
9
+ "image_size": 128,
10
+ "batch_size": 256,
11
+ "num_workers_pgpu": 10,
12
+ "max_epochs": null,
13
+ "warmup_epochs": null,
14
+ "train_steps": 4590,
15
+ "warmup_steps": 500,
16
+ "lr": 0.01,
17
+ "lr_strategy": "one_cycle",
18
+ "num_restarts": null,
19
+ "wd": 0.0001,
20
+ "data_dir": "./data",
21
+ "dataset": "CIFAR10",
22
+ "num_classes": 4,
23
+ "known_classes": [
24
+ 1,
25
+ 0,
26
+ 9,
27
+ 8
28
+ ],
29
+ "osr_dataset": "CIFAR100",
30
+ "osr_classes": [
31
+ 47,
32
+ 6,
33
+ 19,
34
+ 0,
35
+ 62,
36
+ 93,
37
+ 59,
38
+ 65,
39
+ 54,
40
+ 70,
41
+ 34,
42
+ 55,
43
+ 23,
44
+ 38,
45
+ 72,
46
+ 76,
47
+ 53,
48
+ 31,
49
+ 78,
50
+ 96,
51
+ 77,
52
+ 27,
53
+ 92,
54
+ 18,
55
+ 82,
56
+ 50,
57
+ 98,
58
+ 32,
59
+ 1,
60
+ 75,
61
+ 83,
62
+ 4,
63
+ 51,
64
+ 35,
65
+ 80,
66
+ 11,
67
+ 74,
68
+ 66,
69
+ 36,
70
+ 42,
71
+ 33,
72
+ 2,
73
+ 3,
74
+ 97,
75
+ 46,
76
+ 21,
77
+ 64,
78
+ 63,
79
+ 88,
80
+ 43
81
+ ],
82
+ "random_seed": 2023,
83
+ "eval": false,
84
+ "opt": "SGD",
85
+ "save_freq": 50,
86
+ "label_smoothing": 0,
87
+ "sim_block": "attention",
88
+ "num_sim_block": 1,
89
+ "sim_task": "reg",
90
+ "plot_model": false,
91
+ "plot_confusion_matrix": false,
92
+ "plot_feature_embedding": false,
93
+ "plot_roc": false,
94
+ "sim_alpha": 0.01,
95
+ "class_anchor": "mean",
96
+ "anchor_freq": 10,
97
+ "fusion": "cross-attention",
98
+ "model_config": {
99
+ "sim_block": "attention",
100
+ "num_sim_block": 1,
101
+ "sim_task": "reg",
102
+ "patch_size": 16,
103
+ "emb_dim": 768,
104
+ "mlp_dim": 3072,
105
+ "num_heads": 12,
106
+ "sim_num_heads": 12,
107
+ "num_layers": 12,
108
+ "attn_dropout_rate": 0.0,
109
+ "dropout_rate": 0.0
110
+ },
111
+ "tensorboard_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split5_2_exp/tb/",
112
+ "checkpoint_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split5_2_exp/ckpts/",
113
+ "embedding_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split5_2_exp/embeddings/",
114
+ "lightning_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split5_2_exp/lightning_logs/",
115
+ "result_dir": "runs/train/common0527_1_vitb16_CIFAR10+50_split5_2_exp"
116
+ }