add trained_model dir