Skip to content

Retrain the existing model model-32 #8

@yaju1234

Description

@yaju1234

Hello sir,
Thank you for your great work.

I want to re-train your existing model-32 with my custom data set . For that i am replacing
self.load_state_dict(torch.load('../res/resnet50-19c8e357.pth'), strict=False)
with
self.load_state_dict(torch.load('../res/model-32'), strict=False)

But for using that performance of the model is downgraded. Please guide me how to retrain model-32 with my custom data set.
Thanks.

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