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

Handling ConvTranspose2d layers properly#6

Open
eldarkurtic wants to merge 3 commits intoimedslab:masterfrom
eldarkurtic:master
Open

Handling ConvTranspose2d layers properly#6
eldarkurtic wants to merge 3 commits intoimedslab:masterfrom
eldarkurtic:master

Conversation

@eldarkurtic
Copy link

Weights in the Conv2d layer are stored as tensors with shape (out_channels, in_channels, kernel_size[0], kernel_size[1]), while weights in the ConvTranspose2d are stored as tensors with shape (in_channels, out_channels, kernel_size[0], kernel_size[1]).

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