Skip to content

Commit

Permalink
Typo
Browse files Browse the repository at this point in the history
  • Loading branch information
jiexiong2016 committed Jul 5, 2019
1 parent 44f910f commit 63c309e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/GCNextractor.cc
Original file line number Diff line number Diff line change
Expand Up @@ -259,7 +259,7 @@ void GCNextractor::operator()( InputArray _image, InputArray _mask, vector<KeyPo
std::vector<int64_t> dims = {1, img_height, img_width, 1};
auto img_var = torch::from_blob(img.data, dims, torch::kFloat32).to(device);
img_var = img_var.permute({0,3,1,2});
#elif
#else
auto img_tensor = torch::CPU(torch::kFloat32).tensorFromBlob(img.data, {1, img_height, img_width, 1});
img_tensor = img_tensor.permute({0,3,1,2});
auto img_var = torch::autograd::make_variable(img_tensor, false).to(device);
Expand Down

0 comments on commit 63c309e

Please sign in to comment.