i have gotten out of memory errors earlier when the model did not fit, so i dont think that is the issue