Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于微调损失函数 #124

Open
D-W-Y opened this issue May 22, 2024 · 1 comment
Open

关于微调损失函数 #124

D-W-Y opened this issue May 22, 2024 · 1 comment

Comments

@D-W-Y
Copy link

D-W-Y commented May 22, 2024

我发现微调时损失函数只计算分类损失,而不计算重建损失,这是为什么呢?期望您的回答,非常感谢!

@censhallwe
Copy link

微调阶段的训练任务是完成动作分类,不需要重建,是没有解码器参与的,类似于Vision Transformer完成图片分类,只需要用分类损失训练编码器就行。预训练阶段的训练任务是重建RGB帧,编解码器都参与训练,目的是得到一个较优的编码器预训练权重,这样在后续微调阶段用这个预训练好的模型可以更好地完成训练。你也可以看一下其他自监督预训练模型的流程,比如何恺明老师的MAE。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants