add model save in model class
save hidden_probs all epoch