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

预训练 #12

Open
SadInSummer opened this issue May 28, 2024 · 2 comments
Open

预训练 #12

SadInSummer opened this issue May 28, 2024 · 2 comments

Comments

@SadInSummer
Copy link

想请教您一个问题,您提供这个代码目前是在预训练的参数的基础上进行继续训练的对吧,我现在想把这个完全自己训练,然后我把train代码中的预训练的路径改为了None,但是还是一直卡在coyping model for E这个地方

@Lizhe1228
Copy link
Owner

train中加载的权重是Model/Infrared_Visible_Fusion/Infrared/models/10000/pth,是迭代训练后的权重文件,如果想完全自己训练的话,可以把这个文件夹下的所有权重文件删掉。

@SadInSummer
Copy link
Author

我删掉就会卡在 modal for e这一块,等好久都没反应

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