search

ZF4444 / MMAL-Net

This is a PyTorch implementation of the paper "Multi-branch and Multi-scale Attention Learning for Fine-Grained Visual Categorization (MMAL-Net)" (Fan Zhang, Meng Li, Guisheng Zhai, Yizhao Liu).
249 stars 57 forks source link

模型测试需利用到pretrain_model #43

Open Lufei-github opened 10 months ago

Lufei-github commented 10 months ago

你好

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

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

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