如何修改模型

[Muzi010]

作者你好，如果想要修改模型的话应该在哪里修改，感觉无从下手。找不到前向传播的地方，能帮我指出模型的各个部分是在什么地方么。尤其是数据是怎么一步一步的完成整个网络的，麻烦你了。

[lin-tianyu]

你好，

1. 在README中有给出相关文件的位置：important-files-and-folders-to-focus-on
2. 想要读懂代码，需要先了解pytorch lightning的基本工作原理：Pytorch Lightning 1.4.1 Doc，这个库封装的非常顶层，所以代码阅读难度确实较高，这个没办法。
3. 如果你指的是修改里面denoising UNet/autoencoder的网络结构的话，强烈不建议。

[Muzi010]

感谢作者的回答，这个只需对相关损失函数进行修改，或者是对模型的输出进行操作就可以了吧。还有就是对原图进行加噪的话能否去做这个任务，作者进行尝试过么。

[lin-tianyu]

我不是特别理解你的意思。

具体修改模型的什么部分，这个取决于你的项目/你研究的问题呀。要是数学好的话也可以帮忙推推数学，当时开会的时候有人提到这个方法数学上可能有点不完备。

“对原图进行加噪能否做这个任务”，这个是什么意思？举个例子？