Fixes in training reset script