Open hmf opened 2 weeks ago
Hi @hmf; Please correct me if I did not understand correctly :) Here you are mentioning about about the differences in the estimation of output size for the two operations: Conv2d and ConvTranspose2d.
If this is what you mean, thanks for making a note on this. Indeed, this function is designed to target only Conv2d
. Therefore, there might be mismatches in the expected result if you like to get estimation for the ConvTranspose2d
. To handle this issue, this the function needs to be extended further, which is a good idea for the improvement 👍🏼
@tuttelikz
Here you are mentioning about about the differences in the estimation of output size for the two operations: Conv2d and ConvTranspose2d.
That is correct.
To handle this issue, this the function needs to be extended further, which is a good idea for the improvement 👍🏼
To be honest, I though that a simple flip of the matrix width and height would suffice to get the correct output. But this does not seem like it. So, this would be an improvement request.
Alright, qualifying this as feature request then; I will try to keep looking into this on my downtime;
Just a note in case its useful to anyone else.
When trying to calculate the size for a
ConvTranspose2d
, sometimes the estimate is incorrect. Here is an example:The above uses a
ConvTranspose2d
for the real value. I copied and pasted your code and may have made a mistake. Here is the code:and here is the test: