你好，我是新手，请教一个问题

[lechenh]

代码里面的launch_training是训练了该模型，那么请问如何进行预测环节？谢谢。

[Jimmy-7664]

@klaineblue 您好，您是想使用训练好的模型进行预测吗？还是想询问训练过程中在验证集和测试集上预测的部分呢？

[lechenh]

您好，正如您所说的，我希望了解一下后者

[Jimmy-7664]

@klaineblue 好的，那么我想您可以查看这三个runner中的相关代码。
https://github.com/Jimmy-7664/STD-MAE/blob/main/basicts/runners/base_tsf_runner.py （通用的runner）
https://github.com/Jimmy-7664/STD-MAE/blob/main/stdmae/stdmae_runner/mask_runner.py （预训练部分）
https://github.com/Jimmy-7664/STD-MAE/blob/main/stdmae/stdmae_runner/stdmae_runner.py （预测部分）

[lechenh]

十分感谢！

[lechenh]

您好，我在复现过程中注意到您的模型在最初的几轮迭代时损失函数值非常大且一度没有很好的性能提升，请问您当时实验时遇到这种情况是否依然确信自己的模型能够达到卓越的性能。

[Jimmy-7664]

@klaineblue 您好，这是由于课程学习的缘故，模型会逐步学习预测值。
详情可以参考 #22

[lechenh]

感谢您的分享