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.

Sign up for GitHub

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

Jump to bottom

从某个epoch开始接着往后训练 #14

Closed
pumpkin29 opened this issue Dec 29, 2021 · 2 comments
Closed

从某个epoch开始接着往后训练 #14

pumpkin29 opened this issue Dec 29, 2021 · 2 comments

Comments

@pumpkin29
Copy link

您好，请问如果我想从某个epoch开始接着往后训练的话，应该如何修改呢？（比如说，已经训练了40个epoch，但是训练停止了，然后想从第41个epoch开始接着再往后训练20个epoch。）我的想法是：修改config.py，将第40个epoch所得到的权重作为initial_weight读入，将epoch_first_stage设为0，epoch_second_stage设为20。请问这样可以吗
@wenyyu
Copy link
Owner

wenyyu commented Dec 29, 2021

你好，可以的。在config.py 修改args.pre_train 加载预训练模型即可。

@pumpkin29
Copy link
Author

你好，可以的。在config.py 修改args.pre_train 加载预训练模型即可。

好的，谢谢您

@wenyyu wenyyu closed this as completed Dec 29, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@wenyyu @pumpkin29