Hello, I am trying to reproduce your experiment on the MNIST dataset on an optical system. Due to computing power factors, the training process is not smooth. If you can provide the pre-trained weights on the MNIST dataset, it will be of great help to my experiment. Thank you very much.