train a 1024*1024 grayscale image dataset

[linjunhe111]

Hello, thank you for providing this code.
I would like to ask you about training, I want to train a 1024*1024 grayscale image dataset, but I am confused how should I modify the code to train, I hope you can help me with this question! Since I'm new to neural networks, the question posed is rather basic, so please don't mind me
Thank you!