Modifying LaRa Implementation for 256x256 Resolution Reproduction

[intyeger]

Hello,
I'm planning to reproduce your excellent LaRa experiments, but instead of using the 512x512 resolution reported in the paper, I'd like to run everything at 256x256 resolution.
Are there any parts of the code implementation that need to be modified?
Which entries in the configuration files should be updated to reflect the new resolution?

Thank you in advance for your help.

[apchenstu]

I think you can modify the resolution here