Skip to content

模型测试需利用到pretrain_model #43

@Lufei-github

Description

@Lufei-github

你好

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

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

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions