Hugging Face's logo

eulermaxwell
/
AnalogRetriever

PyTorch
English
analog-circuits
cross-modal-retrieval
graph-neural-networks
spice
iccad2026
Model card Files
xet
 Community
eulermax commited on
Commit
51ee703
·
verified ·
1 Parent(s): eb8e52a

Add TI, TIC (GCN), TIC (RGCN) and AnalogRetriever checkpoints

Browse files
Files changed (8) hide show
  1. analogretriever.pth +3 -0
  2. analogretriever.pth.metadata.json +134 -0
  3. ti.pth +3 -0
  4. ti.pth.metadata.json +48 -0
  5. tic_gcn.pth +3 -0
  6. tic_gcn.pth.metadata.json +130 -0
  7. tic_rgcn.pth +3 -0
  8. tic_rgcn.pth.metadata.json +130 -0
analogretriever.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a700f132f426b9cbe28dbf2ddb9fc2d8cb65fe73d16232fba7f6a991f826edc
3
+ size 1764336017
analogretriever.pth.metadata.json ADDED
@@ -0,0 +1,134 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "format_version": 1,
3
+ "model_type": "AnalogRetriever",
4
+ "training_config": {
5
+ "clip_lr": 5e-05,
6
+ "graph_lr": 0.0005,
7
+ "logit_scale_lr": 0.0001,
8
+ "phase1_epochs": 6,
9
+ "phase2_epochs": 2,
10
+ "phase3_epochs": 12,
11
+ "phase1_warmup_ratio": 0.1,
12
+ "phase2_warmup_epochs": 1,
13
+ "alpha_0": 0.05,
14
+ "alpha_max": 0.3,
15
+ "freeze_blocks": 16,
16
+ "label_smoothing": 0.1,
17
+ "batch_size": 256,
18
+ "accum_steps": 1,
19
+ "use_augmentation": true,
20
+ "clip_optimizer_mode": "legacy_cuda_duplicate",
21
+ "seed": 42,
22
+ "clusters": "processed/train_clusters.json",
23
+ "output": "checkpoints/analogretriever.pth"
24
+ },
25
+ "selection": {
26
+ "epoch": 17,
27
+ "directions": [
28
+ "I2C",
29
+ "T2I",
30
+ "T2C"
31
+ ],
32
+ "mean_r1": 75.3499976793925,
33
+ "recall": {
34
+ "I2C": {
35
+ "R@1": 72.84999787807465,
36
+ "R@5": 94.74999845027924,
37
+ "R@10": 97.44999647140503
38
+ },
39
+ "T2I": {
40
+ "R@1": 78.54999721050262,
41
+ "R@5": 96.49999737739563,
42
+ "R@10": 97.94999599456787
43
+ },
44
+ "T2C": {
45
+ "R@1": 74.64999794960022,
46
+ "R@5": 96.84999704360962,
47
+ "R@10": 98.9499968290329
48
+ },
49
+ "C2I": {
50
+ "R@1": 70.74999809265137,
51
+ "R@5": 94.59999918937683,
52
+ "R@10": 97.14999675750732
53
+ },
54
+ "I2T": {
55
+ "R@1": 77.29999721050262,
56
+ "R@5": 96.64999723434448,
57
+ "R@10": 98.24999570846558
58
+ },
59
+ "C2T": {
60
+ "R@1": 71.7499989271164,
61
+ "R@5": 95.34999787807465,
62
+ "R@10": 98.34999680519104
63
+ }
64
+ }
65
+ },
66
+ "source_commit": null,
67
+ "python": "3.12.3",
68
+ "packages": {
69
+ "torch": "2.15.0+cu132",
70
+ "torch-geometric": "2.8.1",
71
+ "torchvision": "0.29.2+cu132",
72
+ "clip": "1.0",
73
+ "numpy": "2.5.4"
74
+ },
75
+ "cuda": "13.2",
76
+ "checkpoint_sha256": "7a700f132f426b9cbe28dbf2ddb9fc2d8cb65fe73d16232fba7f6a991f826edc",
77
+ "graph": {
78
+ "schema_version": 1,
79
+ "node_types": [
80
+ "NET",
81
+ "SUPPLY_NET",
82
+ "GROUND_NET",
83
+ "RESISTOR",
84
+ "CAPACITOR",
85
+ "INDUCTOR",
86
+ "DIODE",
87
+ "V_SOURCE",
88
+ "I_SOURCE",
89
+ "MOSFET",
90
+ "NMOS",
91
+ "PMOS",
92
+ "BJT",
93
+ "NPN",
94
+ "PNP",
95
+ "JFET",
96
+ "NJF",
97
+ "PJF",
98
+ "VCVS",
99
+ "CCCS",
100
+ "VCCS",
101
+ "CCVS",
102
+ "SWITCH",
103
+ "SUBCIRCUIT"
104
+ ],
105
+ "relation_types": [
106
+ "unknown_port",
107
+ "drain",
108
+ "gate",
109
+ "source",
110
+ "bulk",
111
+ "collector",
112
+ "base",
113
+ "emitter",
114
+ "positive",
115
+ "negative",
116
+ "anode",
117
+ "cathode",
118
+ "terminal_1",
119
+ "terminal_2",
120
+ "out_p",
121
+ "out_n",
122
+ "ctrl_in_p",
123
+ "ctrl_in_n",
124
+ "shared_net",
125
+ "subckt_terminal"
126
+ ],
127
+ "continuous_features": [
128
+ "width_um",
129
+ "length_um",
130
+ "resistance_ohm",
131
+ "capacitance_pf"
132
+ ]
133
+ }
134
+ }
ti.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:824da6bd7f83c165a947b59f32cd923cf6b5961258d9b69a9d5bf1056ddbec1c
3
+ size 1710714472
ti.pth.metadata.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "format_version": 1,
3
+ "model_type": "TI",
4
+ "training_config": {
5
+ "clip_lr": 5e-05,
6
+ "logit_scale_lr": 5e-05,
7
+ "epochs": 20,
8
+ "warmup_ratio": 0.1,
9
+ "freeze_blocks": 16,
10
+ "label_smoothing": 0.1,
11
+ "batch_size": 256,
12
+ "accum_steps": 1,
13
+ "use_augmentation": true,
14
+ "clip_optimizer_mode": "legacy_cuda_duplicate",
15
+ "seed": 42,
16
+ "output": "checkpoints/ti.pth"
17
+ },
18
+ "selection": {
19
+ "epoch": 10,
20
+ "directions": [
21
+ "T2I"
22
+ ],
23
+ "mean_r1": 76.79999828338623,
24
+ "recall": {
25
+ "T2I": {
26
+ "R@1": 76.79999828338623,
27
+ "R@5": 95.49999833106995,
28
+ "R@10": 97.8499960899353
29
+ },
30
+ "I2T": {
31
+ "R@1": 76.14999830722809,
32
+ "R@5": 95.69999814033508,
33
+ "R@10": 98.34999561309814
34
+ }
35
+ }
36
+ },
37
+ "source_commit": null,
38
+ "python": "3.12.3",
39
+ "packages": {
40
+ "torch": "2.15.0+cu132",
41
+ "torch-geometric": "2.8.1",
42
+ "torchvision": "0.29.2+cu132",
43
+ "clip": "1.0",
44
+ "numpy": "2.5.4"
45
+ },
46
+ "cuda": "13.2",
47
+ "checkpoint_sha256": "824da6bd7f83c165a947b59f32cd923cf6b5961258d9b69a9d5bf1056ddbec1c"
48
+ }
tic_gcn.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb830c29751a8deda88f1b494c0e949ca488db32bafa033408881780bab72831
3
+ size 1722388045
tic_gcn.pth.metadata.json ADDED
@@ -0,0 +1,130 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "format_version": 1,
3
+ "model_type": "TIC (GCN)",
4
+ "training_config": {
5
+ "conv": "gcn",
6
+ "clip_lr": 5e-05,
7
+ "graph_lr": 0.0005,
8
+ "logit_scale_lr": 0.0001,
9
+ "epochs": 20,
10
+ "phase1_epochs": 6,
11
+ "warmup_ratio": 0.1,
12
+ "freeze_blocks": 16,
13
+ "label_smoothing": 0.1,
14
+ "batch_size": 256,
15
+ "accum_steps": 1,
16
+ "use_augmentation": true,
17
+ "clip_optimizer_mode": "legacy_cuda_duplicate",
18
+ "seed": 42,
19
+ "output": "checkpoints/tic_gcn.pth"
20
+ },
21
+ "selection": {
22
+ "epoch": 15,
23
+ "directions": [
24
+ "I2C",
25
+ "T2I",
26
+ "T2C"
27
+ ],
28
+ "mean_r1": 73.23333164056142,
29
+ "recall": {
30
+ "I2C": {
31
+ "R@1": 69.79999840259552,
32
+ "R@5": 93.39999914169312,
33
+ "R@10": 97.09999680519104
34
+ },
35
+ "T2I": {
36
+ "R@1": 75.6499981880188,
37
+ "R@5": 96.94999694824219,
38
+ "R@10": 98.69999647140503
39
+ },
40
+ "T2C": {
41
+ "R@1": 74.24999833106995,
42
+ "R@5": 96.04999780654907,
43
+ "R@10": 97.74999678134918
44
+ },
45
+ "C2I": {
46
+ "R@1": 67.54999876022339,
47
+ "R@5": 93.79999876022339,
48
+ "R@10": 96.74999713897705
49
+ },
50
+ "I2T": {
51
+ "R@1": 76.3999992609024,
52
+ "R@5": 97.24999666213989,
53
+ "R@10": 98.74999642372131
54
+ },
55
+ "C2T": {
56
+ "R@1": 70.24999916553497,
57
+ "R@5": 95.29999852180481,
58
+ "R@10": 97.5499963760376
59
+ }
60
+ }
61
+ },
62
+ "source_commit": null,
63
+ "python": "3.12.3",
64
+ "packages": {
65
+ "torch": "2.15.0+cu132",
66
+ "torch-geometric": "2.8.1",
67
+ "torchvision": "0.29.2+cu132",
68
+ "clip": "1.0",
69
+ "numpy": "2.5.4"
70
+ },
71
+ "cuda": "13.2",
72
+ "checkpoint_sha256": "fb830c29751a8deda88f1b494c0e949ca488db32bafa033408881780bab72831",
73
+ "graph": {
74
+ "schema_version": 1,
75
+ "node_types": [
76
+ "NET",
77
+ "SUPPLY_NET",
78
+ "GROUND_NET",
79
+ "RESISTOR",
80
+ "CAPACITOR",
81
+ "INDUCTOR",
82
+ "DIODE",
83
+ "V_SOURCE",
84
+ "I_SOURCE",
85
+ "MOSFET",
86
+ "NMOS",
87
+ "PMOS",
88
+ "BJT",
89
+ "NPN",
90
+ "PNP",
91
+ "JFET",
92
+ "NJF",
93
+ "PJF",
94
+ "VCVS",
95
+ "CCCS",
96
+ "VCCS",
97
+ "CCVS",
98
+ "SWITCH",
99
+ "SUBCIRCUIT"
100
+ ],
101
+ "relation_types": [
102
+ "unknown_port",
103
+ "drain",
104
+ "gate",
105
+ "source",
106
+ "bulk",
107
+ "collector",
108
+ "base",
109
+ "emitter",
110
+ "positive",
111
+ "negative",
112
+ "anode",
113
+ "cathode",
114
+ "terminal_1",
115
+ "terminal_2",
116
+ "out_p",
117
+ "out_n",
118
+ "ctrl_in_p",
119
+ "ctrl_in_n",
120
+ "shared_net",
121
+ "subckt_terminal"
122
+ ],
123
+ "continuous_features": [
124
+ "width_um",
125
+ "length_um",
126
+ "resistance_ohm",
127
+ "capacitance_pf"
128
+ ]
129
+ }
130
+ }
tic_rgcn.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff2f3b2e487debcd8e7660fc9ea10f60f12c466cf8d3d66ed9fd5ce7d6369390
3
+ size 1764332650
tic_rgcn.pth.metadata.json ADDED
@@ -0,0 +1,130 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "format_version": 1,
3
+ "model_type": "TIC (RGCN)",
4
+ "training_config": {
5
+ "conv": "rgcn",
6
+ "clip_lr": 5e-05,
7
+ "graph_lr": 0.0005,
8
+ "logit_scale_lr": 0.0001,
9
+ "epochs": 20,
10
+ "phase1_epochs": 6,
11
+ "warmup_ratio": 0.1,
12
+ "freeze_blocks": 16,
13
+ "label_smoothing": 0.1,
14
+ "batch_size": 256,
15
+ "accum_steps": 1,
16
+ "use_augmentation": true,
17
+ "clip_optimizer_mode": "legacy_cuda_duplicate",
18
+ "seed": 42,
19
+ "output": "checkpoints/tic_rgcn.pth"
20
+ },
21
+ "selection": {
22
+ "epoch": 17,
23
+ "directions": [
24
+ "I2C",
25
+ "T2I",
26
+ "T2C"
27
+ ],
28
+ "mean_r1": 74.54999844233195,
29
+ "recall": {
30
+ "I2C": {
31
+ "R@1": 71.8999981880188,
32
+ "R@5": 94.3999993801117,
33
+ "R@10": 96.99999690055847
34
+ },
35
+ "T2I": {
36
+ "R@1": 76.04999840259552,
37
+ "R@5": 97.04999685287476,
38
+ "R@10": 98.14999639987946
39
+ },
40
+ "T2C": {
41
+ "R@1": 75.69999873638153,
42
+ "R@5": 96.34999752044678,
43
+ "R@10": 98.64999651908875
44
+ },
45
+ "C2I": {
46
+ "R@1": 69.19999957084656,
47
+ "R@5": 93.94999921321869,
48
+ "R@10": 96.84999704360962
49
+ },
50
+ "I2T": {
51
+ "R@1": 76.49999797344208,
52
+ "R@5": 96.79999709129333,
53
+ "R@10": 98.49999666213989
54
+ },
55
+ "C2T": {
56
+ "R@1": 72.39999830722809,
57
+ "R@5": 95.59999823570251,
58
+ "R@10": 98.54999721050262
59
+ }
60
+ }
61
+ },
62
+ "source_commit": null,
63
+ "python": "3.12.3",
64
+ "packages": {
65
+ "torch": "2.15.0+cu132",
66
+ "torch-geometric": "2.8.1",
67
+ "torchvision": "0.29.2+cu132",
68
+ "clip": "1.0",
69
+ "numpy": "2.5.4"
70
+ },
71
+ "cuda": "13.2",
72
+ "checkpoint_sha256": "ff2f3b2e487debcd8e7660fc9ea10f60f12c466cf8d3d66ed9fd5ce7d6369390",
73
+ "graph": {
74
+ "schema_version": 1,
75
+ "node_types": [
76
+ "NET",
77
+ "SUPPLY_NET",
78
+ "GROUND_NET",
79
+ "RESISTOR",
80
+ "CAPACITOR",
81
+ "INDUCTOR",
82
+ "DIODE",
83
+ "V_SOURCE",
84
+ "I_SOURCE",
85
+ "MOSFET",
86
+ "NMOS",
87
+ "PMOS",
88
+ "BJT",
89
+ "NPN",
90
+ "PNP",
91
+ "JFET",
92
+ "NJF",
93
+ "PJF",
94
+ "VCVS",
95
+ "CCCS",
96
+ "VCCS",
97
+ "CCVS",
98
+ "SWITCH",
99
+ "SUBCIRCUIT"
100
+ ],
101
+ "relation_types": [
102
+ "unknown_port",
103
+ "drain",
104
+ "gate",
105
+ "source",
106
+ "bulk",
107
+ "collector",
108
+ "base",
109
+ "emitter",
110
+ "positive",
111
+ "negative",
112
+ "anode",
113
+ "cathode",
114
+ "terminal_1",
115
+ "terminal_2",
116
+ "out_p",
117
+ "out_n",
118
+ "ctrl_in_p",
119
+ "ctrl_in_n",
120
+ "shared_net",
121
+ "subckt_terminal"
122
+ ],
123
+ "continuous_features": [
124
+ "width_um",
125
+ "length_um",
126
+ "resistance_ohm",
127
+ "capacitance_pf"
128
+ ]
129
+ }
130
+ }