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