andreasburger commited on
Commit
1012a6b
·
verified ·
1 Parent(s): f0e0319

Delete configs

Browse files
Files changed (1) hide show
  1. configs/equiformer_v2.yaml +0 -57
configs/equiformer_v2.yaml DELETED
@@ -1,57 +0,0 @@
1
- # @package _global_
2
- # from https://github.com/deepprinciple/HORM
3
-
4
- # ocp_trainer: forces_v2
5
-
6
-
7
- model:
8
- name: EquiformerV2
9
-
10
- use_pbc: False
11
- regress_forces: True
12
- otf_graph: True
13
- max_neighbors: 20
14
- max_radius: 12.0
15
- max_num_elements: 90
16
-
17
- num_layers: 4
18
- sphere_channels: 128
19
- attn_hidden_channels: 64 # [64, 96] This determines the hidden size of message passing. Do not necessarily use 96.
20
- num_heads: 4
21
- attn_alpha_channels: 64 # Not used when `use_s2_act_attn` is True.
22
- attn_value_channels: 16
23
- ffn_hidden_channels: 128
24
- norm_type: 'layer_norm_sh' # ['rms_norm_sh', 'layer_norm', 'layer_norm_sh']
25
-
26
- lmax_list: [4]
27
- mmax_list: [2]
28
- grid_resolution: 18 # [18, 16, 14, None] For `None`, simply comment this line.
29
-
30
- num_sphere_samples: 128
31
-
32
- edge_channels: 128
33
- use_atom_edge_embedding: True
34
- share_atom_edge_embedding: False # If `True`, `use_atom_edge_embedding` must be `True` and the atom edge embedding will be shared across all blocks.
35
- distance_function: 'gaussian'
36
- num_distance_basis: 512 # not used
37
-
38
- attn_activation: 'silu'
39
- use_s2_act_attn: False # [False, True] Switch between attention after S2 activation or the original EquiformerV1 attention.
40
- use_attn_renorm: True # Attention re-normalization. Used for ablation study.
41
- ffn_activation: 'silu' # ['silu', 'swiglu']
42
- use_gate_act: False # [True, False] Switch between gate activation and S2 activation
43
- use_grid_mlp: True # [False, True] If `True`, use projecting to grids and performing MLPs for FFNs.
44
- use_sep_s2_act: True # Separable S2 activation. Used for ablation study.
45
-
46
- alpha_drop: 0.1 # [0.0, 0.1]
47
- drop_path_rate: 0.1 # [0.0, 0.05]
48
- proj_drop: 0.0
49
-
50
- weight_init: 'uniform' # ['uniform', 'normal']
51
-
52
- # Added for eigenvalue/eigenvector prediction
53
- do_eigvec_1: true
54
- do_eigvec_2: false
55
- do_eigval_1: true
56
- do_eigval_2: true
57
-