"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "2478cb6f840206eb0d78cd27320d9068be483860"
Commit bc476876 authored by rusty1s's avatar rusty1s
Browse files

typo

parent 9436b36c
...@@ -12,7 +12,7 @@ torch::Tensor fps_cpu(torch::Tensor src, torch::Tensor ptr, double ratio, ...@@ -12,7 +12,7 @@ torch::Tensor fps_cpu(torch::Tensor src, torch::Tensor ptr, double ratio,
CHECK_CPU(src); CHECK_CPU(src);
CHECK_CPU(ptr); CHECK_CPU(ptr);
CHECK_INPUT(ptr.dim() == 1); CHECK_INPUT(ptr.dim() == 1);
AT_ASSERTM(ratio > 0 and ratio < 1, "Invalid input"); AT_ASSERTM(ratio > 0 && ratio < 1, "Invalid input");
src = src.view({src.size(0), -1}).contiguous(); src = src.view({src.size(0), -1}).contiguous();
ptr = ptr.contiguous(); ptr = ptr.contiguous();
......
...@@ -69,7 +69,7 @@ torch::Tensor fps_cuda(torch::Tensor src, torch::Tensor ptr, double ratio, ...@@ -69,7 +69,7 @@ torch::Tensor fps_cuda(torch::Tensor src, torch::Tensor ptr, double ratio,
CHECK_CUDA(src); CHECK_CUDA(src);
CHECK_CUDA(ptr); CHECK_CUDA(ptr);
CHECK_INPUT(ptr.dim() == 1); CHECK_INPUT(ptr.dim() == 1);
AT_ASSERTM(ratio > 0 and ratio < 1, "Invalid input"); AT_ASSERTM(ratio > 0 && ratio < 1, "Invalid input");
cudaSetDevice(src.get_device()); cudaSetDevice(src.get_device());
src = src.view({src.size(0), -1}).contiguous(); src = src.view({src.size(0), -1}).contiguous();
......
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