MMAL-Net icon indicating copy to clipboard operation
MMAL-Net copied to clipboard

模型测试需利用到pretrain_model

Open Lufei-github opened this issue 1 year ago • 0 comments

你好

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

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

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

Lufei-github avatar Jan 08 '24 10:01 Lufei-github