DP答疑交流活動(dòng)第四期-DeePMD-kit模型訓(xùn)練與測(cè)試
2023-07-06 12:12 作者:halfway_right | 我要投稿


訓(xùn)練流程
- 準(zhǔn)備數(shù)據(jù)
- 用dp train命令進(jìn)行訓(xùn)練 (input.json就是上期參數(shù)設(shè)置輸入文件)
- 結(jié)果打印到lcurve.out文件中

如果訓(xùn)練因?yàn)槟承┰蛑袛啵梢允褂胐p train --restart命令重新啟動(dòng)

結(jié)果文件示例
測(cè)試流程

訓(xùn)練完模型以后,可以使用dp freeze和dp test:dp phrase得到完整模型(比如graph.pb 注意.pb文件不是文本文件,看不懂,但是可以進(jìn)行調(diào)用),然后通過dp test命令進(jìn)行調(diào)用該模型,后面指定測(cè)試集數(shù)據(jù)所在路徑

推薦linux gpu(加速計(jì)算)
單個(gè)gpu就行了




標(biāo)簽: