RuntimeError

[yeyulege]

When added the code below, I got an error:" RuntimeError: expected stride to be a single integer value or a list of 3 values to match the convolution dimensions, but got stride=[1, 1] ", does anyone tell me why this happen?

def main():

model = ConvLSTM(input_dim=3,
                 hidden_dim=[64, 64, 128, 128],
                 kernel_size=(3, 3, 3),
                 num_layers=4,
                 batch_first=True,
                 bias=True,
                 return_all_layers=False
                 )

print(model)

x = torch.randn((32, 10, 64, 128, 128))

now_states, last_states = model(x)

if name == "main":
main()

[Jiazxu]

Hi,

Try to set your kernel_size = ([3, 3], [3, 3], [3, 3]).

And reset the "_extend_for_multilayer" method like:

def _extend_for_multilayer(param, num_layers):
        if isinstance(param, int):
            param = [param] * num_layers
        return param

:D

[yeyulege]

Thank you for your advice! That error was solved!