add train

README.md

@@ -2,7 +2,7 @@
  * @Author: zhuww
  * @email: zhuww@sugon.com
  * @Date: 2023-04-15 14:34:07
- * @LastEditTime: 2023-04-01 14:08:01
+ * @LastEditTime: 2023-06-15 10:20:01
 -->
 # ProteinMPNN
 ## 模型介绍
@@ -16,7 +16,7 @@ ProteinMPNN是一种使用MPNN进行蛋白质预测的模型,该模型输入蛋
 ## 推理
 ### 环境配置
 提供[光源](https://www.sourcefind.cn/#/service-details)拉取推理的docker镜像：
-* 推理镜像：docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:proteinmpnn-dtk-22.10-patch4-py37
+* 训练和推理镜像：docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:proteinmpnn-dtk-22.10-patch4-py37
 
 激活镜像环境：
 `source /opt/dtk-22.10/env.sh`
@@ -24,6 +24,16 @@ ProteinMPNN是一种使用MPNN进行蛋白质预测的模型,该模型输入蛋
 测试目录：
 `/opt/ProteinMPNN-main`
 
+### 训练
+训练命令：
+
+    cd /opt/ProteinMPNN-main/training
+    python ./training.py \
+           --path_for_outputs "模型保存路径" \
+           --path_for_training_data "数据集下载并解压路径/pdb_2021aug02" \
+           --num_examples_per_epoch 1个epoch加载的训练example数量 \
+           --save_model_every_n_epochs 每几个epochs保存模型权重
+
 ### 推理
 我们分别提供了基于Pytorch的单体和多体的推理脚本，版本依赖：
 * Pytorch(DCU版本) >= 1.10.0a0
@@ -50,6 +60,7 @@ ProteinMPNN是一种使用MPNN进行蛋白质预测的模型,该模型输入蛋
 | 1 | fp32 | 多体 | 3HTN | 429 | 55.7 |
 | 1 | fp32 | 多体 | 4YOW | 693 | 57.7 |
 
+
 ## 源码仓库及问题反馈
 * https://developer.hpccube.com/codes/modelzoo/ProteinMPNN_Pytorch
 ## 参考