Added try except for model.fit()

      # Add it on to the information needed for the constructor

      hyper_parameters["model_dir"] = model_dir

      model = self.model_builder(**hyper_parameters)

      try:

        model.fit(train_dataset, nb_epoch=nb_epoch)

      # Not all models have nb_epoch

      except TypeError:

        model.fit(train_dataset)

      try:

        model.save()

      # Some models autosave

        model_dir = tempfile.mkdtemp()

      model_params['model_dir'] = model_dir

      model = self.model_builder(**model_params)

      try:

        model.fit(train_dataset, nb_epoch=nb_epoch)

      # Not all models have nb_epoch

      except TypeError:

        model.fit(train_dataset)

      try:

        model.save()

      # Some models autosave