欢迎光临散文网 会员登陆 & 注册

DP答疑交流活动第四期-DeePMD-kit模型训练与测试

2023-07-06 12:12 作者:halfway_right  | 我要投稿

训练流程

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


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



结果文件示例


测试流程

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




推荐linux gpu(加速计算)

单个gpu就行了











DP答疑交流活动第四期-DeePMD-kit模型训练与测试的评论 (共 条)

分享到微博请遵守国家法律