Unverified Commit 4e138bd5 authored by xkszltl's avatar xkszltl Committed by GitHub
Browse files

Add namespace to avoid conflict with ATen version of channel_shuffle(). (#2206)

Fix https://github.com/pytorch/vision/issues/2193.
parent 7a2d0618
...@@ -80,7 +80,7 @@ struct ShuffleNetV2InvertedResidualImpl : torch::nn::Module { ...@@ -80,7 +80,7 @@ struct ShuffleNetV2InvertedResidualImpl : torch::nn::Module {
} else } else
out = torch::cat({branch1->forward(x), branch2->forward(x)}, 1); out = torch::cat({branch1->forward(x), branch2->forward(x)}, 1);
out = channel_shuffle(out, 2); out = ::vision::models::channel_shuffle(out, 2);
return out; return out;
} }
}; };
......
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