fixed param cache problem when training ends