File size: 4,323 Bytes
1416d87 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | {
"model_type": "NEP",
"atom_type": [
8,
72
],
"seed": 2023,
"recover_train": true,
"model": {
"descriptor": {
"cutoff": [
6.0,
6.0
],
"n_max": [
4,
4
],
"basis_size": [
12,
12
],
"l_max": [
4,
2,
1
]
},
"fitting_net": {
"network_size": [
40,
1
]
}
},
"optimizer": {
"optimizer": "ADAM",
"epochs": 30,
"batch_size": 16,
"print_freq": 10,
"learning_rate": 0.001,
"stop_lr": 3.51e-08,
"stop_step": 1000000,
"decay_step": 5000,
"train_energy": true,
"train_force": true,
"train_virial": false,
"start_pre_fac_force": 1000,
"start_pre_fac_etot": 0.02,
"start_pre_fac_virial": 50.0,
"end_pre_fac_force": 1.0,
"end_pre_fac_etot": 1.0,
"end_pre_fac_virial": 1.0
},
"format": "pwmlff/npy",
"train_data": [
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_000_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_002_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_004_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_006_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_008_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_010_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_012_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_014_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_016_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_018_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_020_20/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_022_20/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_024_20/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_026_20/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_001_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_003_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_005_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_007_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_009_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_011_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_013_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_015_30/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_017_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_019_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_021_20/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_023_20/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_025_20/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_027_20/"
],
"valid_data": [
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_000_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_004_50/",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/matpl/HfO2/pwdata/init_008_50/"
]
} |