如何在训练过程中进行调试和修改？

[zhanpengGuan]

您好，我准备在模型上加入新的模块，并预备使用Darts框架（一种基于梯度的神经网络搜索架构）进行训练，在这个过程中将会涉及到不同模块的梯度更新公式的修改，以及对不同模块的相应训练迭代次数的设置，在代码中看来我似乎必须对Pytorch_lightning.trainer进行继承和修改？请问是否能在你们这里得到一些建议？不胜感激！

[Modberge]

您好，根据您的描述，你或许可以参考KGELitModel.py文件的内容，新建一个LitModel文件来自定义你的训练流程，你可以在LitModel类中通过self.current_epoch获取当前训练的迭代次数来选取合适的模块。希望这会对你有所帮助。