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.
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.