Skip to content

Model load fails #3

@antithing

Description

@antithing

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!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions