diff --git a/README.md b/README.md index 8536529..0baf197 100644 --- a/README.md +++ b/README.md @@ -35,8 +35,6 @@ python CGAN.py train 运行上述指令即可开始训练,训练使用 Jittor 框架内置的 MNIST 数据集格式,训练过程中每一个 epoch 的模型系数和结果文件将保存在 `saved/` 文件夹下。 -`CGAN.py` 中 `train()` 函数为用于训练的函数,调用它即可运行训练。相关参数见文件内。 - ### 推理 ```bash @@ -45,8 +43,6 @@ python CGAN.py eval result.png 87 将预训练模型或自行训练的系数 `pkl` 文件放入 `saved/` 文件夹下,并使用上述指令进行推理。推理命令的格式为 `python CGAN.py [output file] [epoch]`,预训练模型 epoch 数为 87,故给出的上述指令使用的是预训练模型推理。 -参见 `CGAN.py` 中的实现。 - ## 致谢 本项目参考了以下资料与项目,感谢作者的分享!