11
11
"graph-ice-au111-bilayer" : "https://zenodo.org/records/10054348/files/weights_ice-au111-bilayer.pth?download=1" ,
12
12
"asdafm-light" : "https://zenodo.org/records/10514470/files/weights_asdafm_light.pth?download=1" ,
13
13
"asdafm-heavy" : "https://zenodo.org/records/10514470/files/weights_asdafm_heavy.pth?download=1" ,
14
+ "edafm-base" : "https://zenodo.org/records/10606273/files/base.pth?download=1" ,
15
+ "edafm-single-channel" : "https://zenodo.org/records/10606273/files/single-channel.pth?download=1" ,
16
+ "edafm-CO-Cl" : "https://zenodo.org/records/10606273/files/CO-Cl.pth?download=1" ,
17
+ "edafm-Xe-Cl" : "https://zenodo.org/records/10606273/files/Xe-Cl.pth?download=1" ,
18
+ "edafm-constant-noise" : "https://zenodo.org/records/10606273/files/constant-noise.pth?download=1" ,
19
+ "edafm-uniform-noise" : "https://zenodo.org/records/10606273/files/uniform_noise.pth?download=1" ,
20
+ "edafm-no-gradient" : "https://zenodo.org/records/10606273/files/no-gradient.pth?download=1" ,
21
+ "edafm-matched-tips" : "https://zenodo.org/records/10606273/files/matched-tips.pth?download=1" ,
14
22
}
15
23
16
24
@@ -21,11 +29,30 @@ def download_weights(weights_name: str, target_path: Optional[PathLike] = None)
21
29
The following weights are available:
22
30
23
31
- ``'graph-ice-cu111'``: PosNet trained on ice clusters on Cu(111). (https://doi.org/10.5281/zenodo.10054348)
24
- - ``'graph-ice-au111-monolayer'``: PosNet trained on monolayer ice clusters on Au(111). (https://doi.org/10.5281/zenodo.10054348)
25
- - ``'graph-ice-au111-bilayer'``: PosNet trained on bilayer ice clusters on Au(111). (https://doi.org/10.5281/zenodo.10054348)
26
- - ``'asdafm-light'``: ASDAFMNet trained on molecules containing the elements H, C, N, O, and F. (https://doi.org/10.5281/zenodo.10514470)
27
- - ``'asdafm-heavy'``: ASDAFMNet trained on molecules additionally containing Si, P, S, Cl, and Br. (https://doi.org/10.5281/zenodo.10514470)
28
-
32
+ - ``'graph-ice-au111-monolayer'``: PosNet trained on monolayer ice clusters on Au(111).
33
+ (https://doi.org/10.5281/zenodo.10054348)
34
+ - ``'graph-ice-au111-bilayer'``: PosNet trained on bilayer ice clusters on Au(111).
35
+ (https://doi.org/10.5281/zenodo.10054348)
36
+ - ``'asdafm-light'``: :class:`.ASDAFMNet` trained on molecules containing the elements H, C, N, O, and F.
37
+ (https://doi.org/10.5281/zenodo.10514470)
38
+ - ``'asdafm-heavy'``: :class:`.ASDAFMNet` trained on molecules additionally containing Si, P, S, Cl, and Br.
39
+ (https://doi.org/10.5281/zenodo.10514470)
40
+ - ``'edafm-base'``: :class:`.EDAFMNet` used for all predictions in the main ED-AFM paper and used for comparison in
41
+ the various tests in the supplementary information of the paper. (https://doi.org/10.5281/zenodo.10606273)
42
+ - ``'edafm-single-channel'``: :class:`.EDAFMNet` trained on only a single CO-tip AFM input.
43
+ (https://doi.org/10.5281/zenodo.10606273)
44
+ - ``'edafm-CO-Cl'``: :class:`.EDAFMNet` trained on alternative tip combination of CO and Cl.
45
+ (https://doi.org/10.5281/zenodo.10606273)
46
+ - ``'edafm-Xe-Cl'``: :class:`.EDAFMNet` trained on alternative tip combination of Xe and Cl.
47
+ (https://doi.org/10.5281/zenodo.10606273)
48
+ - ``'edafm-constant-noise'``: :class:`.EDAFMNet` trained using constant noise amplitude instead of normally distributed
49
+ amplitude. (https://doi.org/10.5281/zenodo.10606273)
50
+ - ``'edafm-uniform-noise'``: :class:`.EDAFMNet` trained using uniform random noise amplitude instead of normally
51
+ distributed amplitude. (https://doi.org/10.5281/zenodo.10606273)
52
+ - ``'edafm-no-gradient'``: :class:`.EDAFMNet` trained without background-gradient augmentation.
53
+ (https://doi.org/10.5281/zenodo.10606273)
54
+ - ``'edafm-matched-tips'``: :class:`.EDAFMNet` trained on data with matched tip distance between CO and Xe,
55
+ instead of independently randomized distances. (https://doi.org/10.5281/zenodo.10606273)
29
56
30
57
Arguments:
31
58
weights_name: Name of weights to download.
0 commit comments