search

happyjin / ConvGRU-pytorch

Convolutional GRU
MIT License
158 stars 29 forks source link

I want to change the output size. #1

Closed kimahiro closed 1 year ago

kimahiro commented 1 year ago

Hi happyjin-san,

Hello there. I am a student from Japan. I tried using your ConvGRU-pytorch implementation.

However, I'm having trouble getting the desired output size even after trying to change the output size of the ConvGRU model. I need the final output size to be (b, t, c, h, w) = (32, 2, 1, 100, 100). Could you please guide me on how to make this modification?

Looking forward to your reply.

happyjin commented 1 year ago

Please refer to https://www.baeldung.com/cs/convolutional-layer-size