Commit 70d30972 authored by yuhai's avatar yuhai
Browse files

Update README.md

parent 09f7d2bd
......@@ -17,7 +17,9 @@ DeePKS-kit架构示意图及工作流程如上图,上图说明了整个迭代
`iterate`: 通过组合上面的四个命令来迭代训练自洽模型
## 环境配置
参考[github链接](https://github.com/deepmodeling/deepks-kit/blob/master/requirements.txt)
DTK版本为22.10 python版本3.7 torch版本1.10.0 pyscf版本2.2.1
## 性能和准确率数据
......@@ -25,72 +27,73 @@ DCU测试平台:Z100
NVIDIA测试平台:A800
**1)收敛性测试**
Deepks的输出结果文件为test.out文件,默认运行的epoch数为10000,从左到右输出的指标分别是训练步骤、训练损失、验证损失、学习率、训练时间、验证时间。其中截取了最后的epoch部分的数值,其中看DCU和A800最后的结果基本一致。
[DCU文件信息]
![](./assets/deepks-1.png)
![](./assets/deepks-2.png)
![](./assets/deepks-3.png)
![](./assets/deepks-4.png)
[A800的文件输出信息]
![](./assets/deepks-5.png)
![](./assets/deepks-6.png)
![](./assets/deepks-7.png)
![](./assets/deepks-8.png)
整体收敛趋势如下图所示,依次为DCU和GPU。
![](./assets/deepks-9.png)
![](./assets/deepks-10.png)
预测值与实际值对照如下,依次为DCU和GPU。
![](./assets/deepks-11.png)
![](./assets/deepks-12.png)
**2)性能测试**
具体性能对比如下表:
| DCU | GPU |
| ------ | ------ |
| cell | cell |
| cell | cell |
![](./assets/deepks-11.png)
## 源码仓库及问题反馈
DeePKS-kit is a pure python library so it can be installed following the standard `git clone` then `pip install` procedure. Note that the two main requirements `pytorch` and `pyscf` will not be installed automatically so you will need to install them manually in advance. Below is a more detailed instruction that includes installing the required libraries in the environment.
![](./assets/deepks-12.png)
We use `conda` here as an example. So first you may need to install [Anaconda](https://docs.anaconda.com/anaconda/install/) or [Miniconda](https://docs.conda.io/en/latest/miniconda.html).
To reduce the possibility of library conflicts, we suggest create a new environment (named `deepks`) with basic dependencies installed (optional):
```bash
conda create -n deepks numpy scipy h5py ruamel.yaml paramiko
conda activate deepks
```
Now you are in the new environment called `deepks`.
Next, install [PyTorch](https://pytorch.org/get-started/locally/)
```bash
# assuming a GPU with cudatoolkit 10.2 support
conda install pytorch cudatoolkit=10.2 -c pytorch
```
and [PySCF](https://github.com/pyscf/pyscf).
```bash
# the conda package does not support python >= 3.8 so we use pip
pip install pyscf
```
**2)性能测试**
Once the environment has been setup properly, using pip to install DeePKS-kit:
```bash
pip install git+https://github.com/deepmodeling/deepks-kit/
```
## Usage
具体性能对比如下表:
An relatively detailed decrisption of the `deepks-kit` library can be found in [here](https://arxiv.org/pdf/2012.14615.pdf). Please also refer to the reference for the description of methods.
Please see [`examples`](./examples) folder for the usage of `deepks-kit` library. A detailed example with executable data for single water molecules can be found [here](./examples/water_single). A more complicated one for training water clusters can be found [here](./examples/water_cluster).
| || DCU | GPU |
| ------ | ------| ------ |
|总训练用时/s|| 23.87 | 27.36 |
|每100epoch用时/s|| 2.38 | 2.73 |
|每100epoch测试/s|| 0.44 | 2.43 |
Check [this input file](./examples/water_cluster/args.yaml) for detailed explanation for possible input parameters, and also [this one](./examples/water_cluster/shell.yaml) if you would like to run on local machine instead of using Slurm scheduler.
## References
## 源码仓库及问题反馈
[1] Chen, Y., Zhang, L., Wang, H. and E, W., 2020. Ground State Energy Functional with Hartree–Fock Efficiency and Chemical Accuracy. The Journal of Physical Chemistry A, 124(35), pp.7155-7165.
https://github.com/deepmodeling/deepks-kit
[2] Chen, Y., Zhang, L., Wang, H. and E, W., 2021. DeePKS: A Comprehensive Data-Driven Approach toward Chemically Accurate Density Functional Theory. Journal of Chemical Theory and Computation, 17(1), pp.170–181.
## 参考
[1] Chen, Y., Zhang, L., Wang, H. and E, W.,2023. DeePKS-kit: A package for developing machine learning-based chemically accurate energy and density functional models, Computer Physics Communications, 282, 108520. https://doi.org/10.1016/j.cpc.2022.108520.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment