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

pretrained model #4

Open
ai25395 opened this issue Apr 17, 2023 · 5 comments
Open

pretrained model #4

ai25395 opened this issue Apr 17, 2023 · 5 comments

Comments

@ai25395
Copy link

ai25395 commented Apr 17, 2023

请问有预训练的模型提供吗,想快速测试一下效果。有的话发个链接可以吗,谢谢。

@ai25395
Copy link
Author

ai25395 commented Apr 17, 2023

训练非常慢,平台是9900k+2080ti,batchsize=1,大约22分钟1个epoch

@CVer-Yang
Copy link
Owner

CVer-Yang commented Apr 17, 2023 via email

@ai25395
Copy link
Author

ai25395 commented Apr 18, 2023

感谢回复。不过这个权重好像和项目里的RCFSNet不匹配,运行test时报错:
RuntimeError: Error(s) in loading state_dict for DataParallel:
E Missing key(s) in state_dict: "module.CDAM2.conv1.weight", .............."module.jiaohu1_Y.layer_norm.bias".

@CVer-Yang
Copy link
Owner

CVer-Yang commented Apr 18, 2023

这个模型发表后为了便于理解修改了一些函数名字,可能导致模型权重不对应,我已上传一版原始模型文件(networks->RCFSNet-V1.py),可以实现权重对应,该文件内容较为冗余, 但主要部分与思想一致。

@axin1301
Copy link

Your work is really impressive. Can you please provide the pre-trained weights on the DeepGlobal dataset? Thank you.

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

3 participants