DeePMD-kit训练模型的用法
在程序包的examples文件夹里多个算例,路径如下:
单核运行的话,可以直接执行下面的命令:
这里input.json就是输入文件的文件名。默认情况下,DeePMD-KIT的详细级别为INFO,屏幕上有关代码和环境的重要信息有许多。需要特别关注关于数据系统的两条信息。
DeePMD-KIT会打印关于训练和验证数据集的详细信息,由输入脚本中“training”部分的“Training_Data”和“Validation_Data”定义。训练数据集由三个数据系统组成,而验证数据集由一个数据系统组成。系统中的原子数量、batch_size、批次数量以及使用该系统的概率都显示在屏幕上。最后一列显示系统是否假定为周期性边界条件。
在训练过程中,用于训练模型的批次和验证数据的number_bch批次的每个disp_freq训练步都开展模型误差测试。训练误差和验证误差打印在disp_file文件中(默认为lcurve.out)。batch_size可以输入脚本中关于训练和验证数据集部分的关键字BATCH_SIZE来设置。