cyrusyc commited on
Commit
d78049e
1 Parent(s): 05e8129

add package and ckpt info

Browse files
Files changed (1) hide show
  1. mlip_arena/models/registry.yaml +18 -0
mlip_arena/models/registry.yaml CHANGED
@@ -2,6 +2,8 @@ MACE-MP(M):
2
  module: externals
3
  class: MACE_MP_Medium
4
  family: mace-mp
 
 
5
  username: cyrusyc
6
  last-update: 2024-03-25T14:30:00
7
  datetime: 2024-03-25T14:30:00 # TODO: Fake datetime
@@ -25,6 +27,8 @@ CHGNet:
25
  module: externals
26
  class: CHGNet
27
  family: chgnet
 
 
28
  username: cyrusyc
29
  last-update: 2024-07-08T00:00:00
30
  datetime: 2024-07-08T00:00:00
@@ -45,6 +49,8 @@ M3GNet:
45
  module: externals
46
  class: M3GNet
47
  family: m3gnet
 
 
48
  username: cyrusyc
49
  last-update: 2024-07-08T00:00:00
50
  datetime: 2024-07-08T00:00:00
@@ -64,6 +70,8 @@ EquiformerV2(OC22):
64
  module: externals
65
  class: EquiformerV2
66
  family: equiformer
 
 
67
  username: cyrusyc
68
  last-update: 2024-07-08T00:00:00
69
  datetime: 2024-07-08T00:00:00
@@ -83,6 +91,8 @@ EquiformerV2(OC20):
83
  module: externals
84
  class: EquiformerV2OC20
85
  family: equiformer
 
 
86
  username: cyrusyc
87
  last-update: 2024-07-08T00:00:00
88
  datetime: 2024-07-08T00:00:00
@@ -102,6 +112,8 @@ eSCN(OC20):
102
  module: externals
103
  class: eSCN
104
  family: escn
 
 
105
  username: cyrusyc
106
  last-update: 2024-07-08T00:00:00
107
  datetime: 2024-07-08T00:00:00
@@ -142,6 +154,8 @@ ALIGNN:
142
  module: externals
143
  class: ALIGNN
144
  family: alignn
 
 
145
  username: cyrusyc
146
  last-update: 2024-07-08T00:00:00
147
  datetime: 2024-07-08T00:00:00
@@ -161,6 +175,8 @@ SevenNet:
161
  module: externals
162
  class: SevenNet
163
  family: sevennet
 
 
164
  username: cyrusyc
165
  last-update: 2024-03-25T14:30:00
166
  datetime: 2024-03-25T14:30:00 # TODO: Fake datetime
@@ -183,6 +199,8 @@ ORB:
183
  module: externals
184
  class: ORB
185
  family: orb
 
 
186
  username: cyrusyc
187
  last-update: 2024-03-25T14:30:00
188
  datetime: 2024-03-25T14:30:00 # TODO: Fake datetime
 
2
  module: externals
3
  class: MACE_MP_Medium
4
  family: mace-mp
5
+ package: mace-torch==0.3.4
6
+ checkpoint: 2023-12-03-mace-128-L1_epoch-199.model
7
  username: cyrusyc
8
  last-update: 2024-03-25T14:30:00
9
  datetime: 2024-03-25T14:30:00 # TODO: Fake datetime
 
27
  module: externals
28
  class: CHGNet
29
  family: chgnet
30
+ package: chgnet==0.3.5
31
+ checkpoint:
32
  username: cyrusyc
33
  last-update: 2024-07-08T00:00:00
34
  datetime: 2024-07-08T00:00:00
 
49
  module: externals
50
  class: M3GNet
51
  family: m3gnet
52
+ package: matgl==1.1.2
53
+ checkpoint:
54
  username: cyrusyc
55
  last-update: 2024-07-08T00:00:00
56
  datetime: 2024-07-08T00:00:00
 
70
  module: externals
71
  class: EquiformerV2
72
  family: equiformer
73
+ package: fairchem-core==1.0.0
74
+ checkpoint: EquiformerV2-lE4-lF100-S2EFS-OC22
75
  username: cyrusyc
76
  last-update: 2024-07-08T00:00:00
77
  datetime: 2024-07-08T00:00:00
 
91
  module: externals
92
  class: EquiformerV2OC20
93
  family: equiformer
94
+ package: fairchem-core==1.0.0
95
+ checkpoint: EquiformerV2-31M-S2EF-OC20-All+MD
96
  username: cyrusyc
97
  last-update: 2024-07-08T00:00:00
98
  datetime: 2024-07-08T00:00:00
 
112
  module: externals
113
  class: eSCN
114
  family: escn
115
+ package: fairchem-core==1.0.0
116
+ checkpoint: eSCN-L6-M3-Lay20-S2EF-OC20-All+MD
117
  username: cyrusyc
118
  last-update: 2024-07-08T00:00:00
119
  datetime: 2024-07-08T00:00:00
 
154
  module: externals
155
  class: ALIGNN
156
  family: alignn
157
+ package: alignn==2024.5.27
158
+ checkpoint:
159
  username: cyrusyc
160
  last-update: 2024-07-08T00:00:00
161
  datetime: 2024-07-08T00:00:00
 
175
  module: externals
176
  class: SevenNet
177
  family: sevennet
178
+ package: sevenn==0.9.4
179
+ checkpoint: 7net-0
180
  username: cyrusyc
181
  last-update: 2024-03-25T14:30:00
182
  datetime: 2024-03-25T14:30:00 # TODO: Fake datetime
 
199
  module: externals
200
  class: ORB
201
  family: orb
202
+ package: orb-models==0.3.1
203
+ checkpoint: orb_v1
204
  username: cyrusyc
205
  last-update: 2024-03-25T14:30:00
206
  datetime: 2024-03-25T14:30:00 # TODO: Fake datetime