DP答疑交流活动第四期-DeePMD-kit模型训练与测试
2023-07-06 12:12 作者:halfway_right | 我要投稿


训练流程
- 准备数据
- 用dp train命令进行训练 (input.json就是上期参数设置输入文件)
- 结果打印到lcurve.out文件中

如果训练因为某些原因中断,可以使用dp train --restart命令重新启动

结果文件示例
测试流程

训练完模型以后,可以使用dp freeze和dp test:dp phrase得到完整模型(比如graph.pb 注意.pb文件不是文本文件,看不懂,但是可以进行调用),然后通过dp test命令进行调用该模型,后面指定测试集数据所在路径

推荐linux gpu(加速计算)
单个gpu就行了



