|
@parameterized( |
|
("concat", "strip"), # repinit_tebd_input_mode |
|
(True,), # repinit_set_davg_zero |
|
(False,), # repinit_type_one_side |
|
(True, False), # repinit_use_three_body |
|
(True, False), # repformer_direct_dist |
|
(True,), # repformer_update_g1_has_conv |
|
(True,), # repformer_update_g1_has_drrd |
|
(True,), # repformer_update_g1_has_grrg |
|
(True,), # repformer_update_g1_has_attn |
|
(True,), # repformer_update_g2_has_g1g1 |
|
(True,), # repformer_update_g2_has_attn |
|
(False,), # repformer_update_h2 |
|
(True,), # repformer_attn2_has_gate |
|
("res_avg", "res_residual"), # repformer_update_style |
|
("norm", "const"), # repformer_update_residual_init |
|
(True,), # repformer_set_davg_zero |
|
(True,), # repformer_trainable_ln |
|
(1e-5,), # repformer_ln_eps |
|
(True,), # repformer_use_sqrt_nnei |
|
(True,), # repformer_g1_out_conv |
|
(True,), # repformer_g1_out_mlp |
|
(True, False), # smooth |
|
([], [[0, 1]]), # exclude_types |
|
("float64",), # precision |
|
(True, False), # add_tebd_to_repinit_out |
|
(True, False), # use_econf_tebd |
|
(False,), # use_tebd_bias |
|
) |
Currently, we have 1024 variations for DPA-1 and 512 variations for DPA-2, which takes a lot of CI time, especially when we add more supported backends.
deepmd-kit/source/tests/consistent/descriptor/test_dpa1.py
Lines 42 to 62 in ad8bebe
deepmd-kit/source/tests/consistent/descriptor/test_dpa2.py
Lines 44 to 72 in ad8bebe
cc @wanghan-iapcm @iProzd