-
Notifications
You must be signed in to change notification settings - Fork 2
Description
Hi! and thank you for making this code available. i am building on Windows, with libtorch 2.0.0 / cuda 11.7.
The code compiles after I edit a few places, but i see a crash when loading the model you provide:
model = make_shared<SuperPoint>();
torch::load(model, str_model); //crash here
The edits I made are:
auto fkpts = torch::from_blob(kpt_mat.data, {l**ong long**(keypoints.size()), 2}, torch::kFloat); //to get round a narrowing conversion error
and:
auto desc = torch::grid_sampler(mDesc, grid, 0, 0, **false**); // [1, 256, 1, n_keypoints] //to get rid of a 'invalid args' error
None of these should effect the model load though! What might be happening here?
EDIT: i see teh same crash when loading teh gcn2 model. On this line:
module = std::make_shared<torch::jit::script::Module>(torch::jit::load(net_fn));
Both of these have been tested with hardcoded and double checked model paths.
Thank you!