Skip to content

GitLab

Unverified Commit f02caa53 authored by Duc Nguyen's avatar Duc Nguyen Committed by GitHub
Browse files

Fix ShuffleNet ONNX export (#7686) 


Co-authored-by: default avatarNicolas Hug <nh.nicolas.hug@gmail.com>
parent b5401b94
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/models/shufflenetv2.py
View file @ f02caa53
...@@ -35,7 +35,7 @@ def channel_shuffle(x: Tensor, groups: int) -> Tensor: ...@@ -35,7 +35,7 @@ def channel_shuffle(x: Tensor, groups: int) -> Tensor:
x = torch.transpose(x, 1, 2).contiguous() x = torch.transpose(x, 1, 2).contiguous()
# flatten # flatten
x = x.view(batchsize, -1, height, width) x = x.view(batchsize, num_channels, height, width)
return x return x
......
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