Bigger input images

[andresherrera97]

Hello, I'm trying to use this model to predict cloud movement on 256x256 images. I want to base the model on the one you used for SST but I'm not sure I can replicate it.

Which values did you use here and could you recommend me what would you use for 256x256 images?

phycell = PhyCell(input_shape=(64,64), input_dim=64, F_hidden_dims=[49], n_layers=1, kernel_size=(7,7), device=device)

convcell = ConvLSTM(input_shape=(64,64), input_dim=64, hidden_dims=[128,128,64], n_layers=3, kernel_size=(3,3), device=device)

Are there other params of the architecture I should change?

Thanks for the code, I think it looks very promising for what I'm trying to do