You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi All,
I'm now confused with a problem of forwarding the model in NCNN.
The model is resnet50 trained under PyTorch1.0. I delete all the keys with "num_batches_tracked"of the state_dict created by the code:torch.load(model_path).
It succeeded in converting the model, but failed when run:
in my C++ codes. exis a ncnn::Extractor, which can successfully extract all the layers' outputs except the last layer:Addmm_1. Addmm_1 is converted from:
Hi All,
I'm now confused with a problem of forwarding the model in NCNN.
The model is resnet50 trained under PyTorch1.0. I delete all the keys with
"num_batches_tracked"
of thestate_dict
created by the code:torch.load(model_path)
.It succeeded in converting the model, but failed when run:
in my C++ codes.
ex
is ancnn::Extractor
, which can successfully extract all the layers' outputs except the last layer:Addmm_1
.Addmm_1
is converted from:in my Python codes.
Anyone have ever suffered the same problem?
The text was updated successfully, but these errors were encountered: