Resizing images in sth and small fix, trainer uses the solver file if it exists
New architecture for the synth dataset