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

请问下怎么使用你最新的预训练模型呢? #5

Open
xtanitfy opened this issue Nov 10, 2018 · 4 comments
Open

请问下怎么使用你最新的预训练模型呢? #5

xtanitfy opened this issue Nov 10, 2018 · 4 comments

Comments

@xtanitfy
Copy link

我直接使用你的代码和数据,训练了一晚上精度只有70%,后边精度突然变为了0,我怀疑是预训练的模型没有记载你最新的模型,请问我该怎么做呢?多谢你分享的源码!

@xtanitfy
Copy link
Author

可能是因为我训练的时候断掉了网络导致没有下载预训练模型的原因,我再试试看

@yinchangchang
Copy link
Owner

先加载torchvision的densenet预训练模型,这样会有一个良好的初始化参数,让网络更容易训练的

@xtanitfy
Copy link
Author

请问训练求loss的时候,你的每个8×8的小方格的标签是怎么打上并用分类的交叉熵求分类损失的?

@yinchangchang
Copy link
Owner

最后在输出的label上,用了一个max pooling,就可以算loss了

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