Skip to content
This repository was archived by the owner on Apr 18, 2026. It is now read-only.

fix BottleNeckBlock skip_connect bug#48

Open
n3011 wants to merge 1 commit intogoogle:masterfrom
n3011:master
Open

fix BottleNeckBlock skip_connect bug#48
n3011 wants to merge 1 commit intogoogle:masterfrom
n3011:master

Conversation

@n3011
Copy link
Copy Markdown

@n3011 n3011 commented Mar 28, 2021

This PR fixes the bug related to skip_connet in the BottleNeckBlock block.

Instead of using the flattened tensor for getting the original dimension and adding with the output tensor, the original tensor was used, which will cause a shape mismatch error for an original input tensor with more than 2 dimensions.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant