harikrishnanvs comments

Repositories
Issues
Comments

Results 1 comments of


                                            harikrishnanvs

How to do inference using this model?

Hi, @Akhiladdh Solved the same error that happened in dense_localization.py by changing line "model.load_state_dict(torch.load("model/DenseNet121_aug4_pretrain_WeightBelow1_1_0.829766922537.pkl"))" to state_dict =torch.load("model/DenseNet121_aug4_pretrain_WeightBelow1_1_0.829766922537.pkl") model.load_state_dict(state_dict,strict=False) But, Now a new error raised. generate heatmap .......... Traceback (most recent...