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)簽：
