Hi ChiaWei,
I understand that you are trying to resolve the out of memory error raised while training an autoencoder in MATLAB.
- The standard method of resolving this error is to reduce the size of training images. However, as you are looking for other ways of tackling the issue, one way is to try decreasing the “hiddenSize” parameter so that the number of learnable parameters of the model reduces, thereby decreasing the need for higher memory requirement.
- You can also try decreasing the parameter “MaxEpochs”.
- Also, try to increase the GPU resources allocated for training the model.
Please refer to the below documentation to learn more about the different parameters that can be tweaked in the autoencoder model: https://www.mathworks.com/help/deeplearning/ref/trainautoencoder.html
Hope it helps.
Regards,
Sai Pavan
