fix sponge readme

model_zoo/research/hpc/sponge/README.md

@@ -16,11 +16,11 @@
 
 SPONGE in MindSpore is a high-performance and modularized molecular dynamics simulation library developed by `Yiqin Gao` group (`Peking University` and `Shenzhen Bay Laboratory`) and `MindSpore` team at `Huawei` Company. It can efficiently simulate traditional molecular dynamics tasks based on the “graph-kernel-fusion” and “automatic parallelization” features of MindSpore. In the meanwhile, it utilizes the automatic differentiation feature of MindSpore, and introduces machine learning methods, such as neural network, into traditional molecular simulation, achieving methodological inventions.
 
-This example demonstrates how to perform high-performance molecular dynamics simulations with the built-in SPONGE module of MindSpore on GPU.
+This example demonstrates how to perform high-performance molecular dynamics simulations with the built-in SPONGE module of MindSpore on `GPU`.
 
 ## Dataset
 
-There are three inputs for the example, property file `NVT_290_10ns.in`, topology file `ala.parm7` and coordinates file `ala_NVT_290_10ns.out`, respectivelly.
+There are three inputs for the example, property file `NVT_290_10ns.in`, topology file `WATER_ALA.parm7` and coordinates file `WATER_ALA_350_cool_290.rst7`, respectivelly.
 
 ![ALA Aqueous System](https://images.gitee.com/uploads/images/2021/0323/184453_4bd9b1a6_8142020.png "图片1.png")
 
@@ -30,8 +30,8 @@ Topology file and coordinates file can be generated by `tleap` in `AmberTools` (
 
 ## Environment Requirements
 
-- Hardware: GPU
-    - Prepare hardware environment with GPU processor.
+- Hardware: `GPU`
+    - Prepare hardware environment with `GPU` processor.
 - Framework
     - [MindSpore](https://www.mindspore.cn/install/en)
 - For more information, please check the resources below:
@@ -43,7 +43,9 @@ Topology file and coordinates file can be generated by `tleap` in `AmberTools` (
 After installing MindSpore via the official website, you can start running as follows:
 
 ```shell
-python main.py --i /path/NVT_290_10ns.in --amber_parm /path/ala.parm7 --c /path/ala_350_cool_290.rst7 \
+python main.py --i /path/NVT_290_10ns.in \
+               --amber_parm /path/WATER_ALA.parm7 \
+               --c /path/WATER_ALA_350_cool_290.rst7 \
                --o /path/ala_NVT_290_10ns.out
 ```
 
@@ -67,13 +69,16 @@ python main.py --i /path/NVT_290_10ns.in --amber_parm /path/ala.parm7 --c /path/
         ├── md_information.py                    # save md information module in SPONGE
         ├── neighbor_list.py                     # neighbor_list module in SPONGE
         ├── particle_mesh_ewald.py               # particle_mesh_ewald module in SPONGE
-        ├── simulation_initial.py                # SPONGE simulation
+        ├── simulation.py                # SPONGE simulation
 ```
 
 ### Training Process
 
 ```shell
-python main.py --i ./NVT_290_10ns.in --amber_parm ala.parm7 --c ala_350_cool_290.rst7 --o ala_NVT_290_10ns.out
+python main.py --i /path/NVT_290_10ns.in \
+               --amber_parm /path/WATER_ALA.parm7 \
+               --c /path/WATER_ALA_350_cool_290.rst7 \
+               --o /path/ala_NVT_290_10ns.out
 ```
 
 Training result will be stored in the specified file, which ends with ".out".
@@ -101,8 +106,8 @@ There are sorts of energy in the output, steps (_steps_), temperature (_TEMP_),
 | MindSpore Version          | 1.2
 | Training Parameters        | step=1
 | Outputs                    | numpy file
-| Speed                      | 0.47 s/step
-| Total time                 | 4.57 s
+| Speed                      | 16.7 ms/step
+| Total time                 | 10.7 s
 | Scripts                    | [Link](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/hpc/sponge)
 
 ## ModelZoo HomePage