Thanks for sharing the code.
Your current model input is the initial condition and hidden states for the ConvLSTM,
From a perspective of generalization, have you considered feeding a batch of initial conditions, instead of a single one to train the model? Would that increase the performance?
Also, does a single initial condition in PICRNN equal to a single collocation point in PINN? Not sure how to justify that.
Thanks for sharing the code. Your current model input is the initial condition and hidden states for the ConvLSTM, From a perspective of generalization, have you considered feeding a batch of initial conditions, instead of a single one to train the model? Would that increase the performance? Also, does a single initial condition in PICRNN equal to a single collocation point in PINN? Not sure how to justify that.
Thanks again for your work.