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
wants to merge 3 commits into
base: master
Choose a base branch
from

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