set save state after training, another nham/nspam mixup i forgot to fix.