Akhila comments

Repositories
Issues
Comments

Results 1 comments of


                                            Akhila

How to do inference using this model?

I am getting below error if I do local testing `Traceback (most recent call last): File "denseNet_localization.py", line 78, in model.load_state_dict(torch.load("model/DenseNet121_aug4_pretrain_WeightBelow1_1_0.829766922537.pkl")) File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 769, in load_state_dict self.__class__.__name__, "\n\t".join(error_msgs))) RuntimeError:...