DeePMD专题

DeePMD DPA4C紧凑描述符训练与测试

列出实际模型输入,读取学习曲线与独立测试构型的能量受力误差。

更新于 2026-11-11约 2 分钟阅读按任务与环境选择适用配置
阅读本篇目录

本篇接着输入与环境准备,读取DPA4C紧凑描述符的新训练结果。下载完整实例包。

1 打开完整训练输入#

cat inputs/dpa4c.json

实际model部分如下;完整文件还给出学习率、损失及训练/验证数据路径:

{
  "type": "standard",
  "type_map": [
    "N",
    "H"
  ],
  "descriptor": {
    "type": "dpa4c",
    "rcut": 3.0,
    "channels": 16,
    "lmax": 2,
    "n_radial": 4,
    "radial_modes": 0,
    "use_amp": false,
    "precision": "float64",
    "seed": 600001
  },
  "fitting_net": {
    "type": "ener",
    "neuron": [
      32,
      32
    ],
    "resnet_dt": false,
    "precision": "float64",
    "seed": 600002
  }
}

channels=16、lmax=2、n_radial=4控制描述符的通道、角向与径向规模;radial_modes=0是本例采用的设置。precision=float64便于比较能量导数,use_amp=false关闭混合精度。DPA4C使用标准能量拟合网络,不需要DPA4的专用model.type结构;这两个入口不能只改一个名字就互换。

本例batch 4,训练3000步,起始学习率0.002、终值0.00001。总能量与力损失权重均为1,virial权重为0。力来自总能量对坐标的导数。固定NH₃化学计量下的总能量监督不能确定唯一的物理原子能量分解。

2 查看自己的误差表#

head -n 3 tables-new/model-results.csv

这里有自己新训练模型的一行结果;能量按4个真实原子归一,力为12个测试姿态中全部原子三方向的分量RMSE。本站P100与V10016G独立训练的实际结果如下:

实算设备 能量RMSE(eV/原子) 力RMSE(eV/Å)
P100 0.0002790 0.026628
V100 0.0001103 0.020393

heldout-energy-and-forces

不同设备从头训练的优化轨迹可能不同;后面的CPU/GPU检查会对比同一个权重,两种比较不要混淆。小样例只覆盖固定氨分子构型范围,不据此宣称通用反应势、解离精度或生产MD稳定性。

3 读取学习曲线#

head -n 6 new-work/p100-dpa4c/lcurve.out

V100运行时将目录前缀改为v100-。日志区分训练和验证误差,测试构型不参与优化;不要只看最后一个训练batch来判断模型能否外推。

native-learning-curves

4 查看本次作业摘要#

python3 code/summary.py

应看到1个新模型、3000次优化更新和新结果检查摘要,新量子化学调用为0。新的训练日志、测试预测与推断数组在new-work,附带参考在reference;两者分开保存。

需要根据你的体系选择资源?联系讨论计算需求。实际可用软件与权限,以当前账户和运行环境为准。

查看配图

放大后可滚动查看细节,按 Esc 关闭。

100%查看原图 ↗