Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "a27755db416644273de8d70b770f1d2e68db720b"
Unverified Commit 90c6a951 authored by Hang Zhang's avatar Hang Zhang Committed by GitHub
Browse files

1.4.0 patch (#246)

parent ae218bf9
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
encoding/lib/cpu/nms_cpu.cpp
View file @ 90c6a951
...@@ -32,8 +32,8 @@ std::vector<at::Tensor> Non_Max_Suppression_CPU( ...@@ -32,8 +32,8 @@ std::vector<at::Tensor> Non_Max_Suppression_CPU(
AT_ASSERT(input.size(2) == 4); AT_ASSERT(input.size(2) == 4);
AT_ASSERT(input.is_contiguous()); AT_ASSERT(input.is_contiguous());
AT_ASSERT(scores.is_contiguous()); AT_ASSERT(scores.is_contiguous());
AT_ASSERT(input.type().scalarType() == at::kFloat || input.type().scalarType() == at::kDouble) AT_ASSERT(input.type().scalarType() == at::kFloat || input.type().scalarType() == at::kDouble);
AT_ASSERT(scores.type().scalarType() == at::kFloat || scores.type().scalarType() == at::kDouble) AT_ASSERT(scores.type().scalarType() == at::kFloat || scores.type().scalarType() == at::kDouble);
AT_ASSERT(input.is_contiguous()); AT_ASSERT(input.is_contiguous());
AT_ASSERT(scores.is_contiguous()); AT_ASSERT(scores.is_contiguous());
......
encoding/lib/gpu/nms_kernel.cu
View file @ 90c6a951
...@@ -71,8 +71,8 @@ std::vector<at::Tensor> Non_Max_Suppression_CUDA( ...@@ -71,8 +71,8 @@ std::vector<at::Tensor> Non_Max_Suppression_CUDA(
AT_ASSERT(input.size(2) == 4); AT_ASSERT(input.size(2) == 4);
AT_ASSERT(input.is_contiguous()); AT_ASSERT(input.is_contiguous());
AT_ASSERT(scores.is_contiguous()); AT_ASSERT(scores.is_contiguous());
AT_ASSERT(input.type().scalarType() == at::kFloat || input.type().scalarType() == at::kDouble) AT_ASSERT(input.type().scalarType() == at::kFloat || input.type().scalarType() == at::kDouble);
AT_ASSERT(scores.type().scalarType() == at::kFloat || scores.type().scalarType() == at::kDouble) AT_ASSERT(scores.type().scalarType() == at::kFloat || scores.type().scalarType() == at::kDouble);
auto num_boxes = input.size(1); auto num_boxes = input.size(1);
auto batch_size = input.size(0); auto batch_size = input.size(0);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment