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

模型测试需利用到pretrain_model #43

Open

Lufei-github opened this issue Jan 8, 2024 · 0 comments

Lufei-github commented Jan 8, 2024

你好

我发现测试代码首先会在test.py中model = MainNet()处加载预训练模型 (config.py中pretrain_path = './models/pretrained/resnet50-19c8e357.pth'）

然后会在test.py中的epoch = auto_load_resume(model, pth_path, status='test')，又加载一次你训练的模型

那可不可以只加载最终的模型，而不用加载预训练模型呢？

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment