|
|
|
@@ -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` 中的实现。 |
|
|
|
|
|
|
|
## 致谢 |
|
|
|
|
|
|
|
本项目参考了以下资料与项目，感谢作者的分享！ |
|
|
|
|
KSkun
3 years ago