Example #1
0
def main(argv):
  if len(argv) > 1:
    raise tf.app.UsageError("Too many command-line arguments.")

  cfg1_text = compare_params.get_model_params_as_text(FLAGS.model1)
  cfg2_text = compare_params.get_model_params_as_text(FLAGS.model2)

  cfg1_not_cfg2, cfg2_not_cfg1, cfg1_and_cfg2_diff = (
      compare_params.hyperparams_text_diff(cfg1_text, cfg2_text))

  compare_params.print_hyperparams_text_diff(FLAGS.model1, FLAGS.model2,
                                             cfg1_not_cfg2, cfg2_not_cfg1,
                                             cfg1_and_cfg2_diff)
  return 0
  def testBasic(self):
    h1 = hyperparams.Params()
    h1.Define('a', 1, '')
    h1.Define('b', 2, '')

    h2 = hyperparams.Params()
    h2.Define('a', 3, '')  # different value.
    h2.Define('c', 2, '')

    d1, d2, d3 = compare_params.hyperparams_text_diff(h1.ToText(), h2.ToText())
    self.assertEqual(d1, ['b'])
    self.assertEqual(d2, ['c'])
    self.assertEqual(d3, {'a': ('1', '3')})

    # Exercise print function
    compare_params.print_hyperparams_text_diff('h1', 'h2', d1, d2, d3)