Python populate_experiment Beispiele

Beispiel #1

 def test_experiment(self):
     """Tests that the `Experiment` object is constructed correctly."""
     run_config = tf.estimator.RunConfig()
     hparams = model_hparams.create_hparams(
         hparams_overrides='load_pretrained=false')
     pipeline_config_path = get_pipeline_config_path(MODEL_NAME_FOR_TEST)
     experiment = model_lib.populate_experiment(run_config,
                                                hparams,
                                                pipeline_config_path,
                                                train_steps=10,
                                                eval_steps=20)
     self.assertEqual(10, experiment.train_steps)
     self.assertEqual(20, experiment.eval_steps)

Beispiel #2

Datei: model_lib_test.py Projekt: ALISCIFP/models

 def test_experiment(self):
   """Tests that the `Experiment` object is constructed correctly."""
   run_config = tf.estimator.RunConfig()
   hparams = model_hparams.create_hparams(
       hparams_overrides='load_pretrained=false')
   pipeline_config_path = get_pipeline_config_path(MODEL_NAME_FOR_TEST)
   experiment = model_lib.populate_experiment(
       run_config,
       hparams,
       pipeline_config_path,
       train_steps=10,
       eval_steps=20)
   self.assertEqual(10, experiment.train_steps)
   self.assertEqual(20, experiment.eval_steps)