Python list_ranged_hparams Exemples

Exemple #1

    def testListHparams(self):
        @registry.register_hparams
        def hp1():
            pass

        @registry.register_hparams("hp2_named")
        def hp2():
            pass

        @registry.register_ranged_hparams
        def rhp1(_):
            pass

        @registry.register_ranged_hparams("rhp2_named")
        def rhp2(_):
            pass

        self.assertSetEqual(set(["hp1", "hp2_named"]),
                            set(registry.list_hparams()))
        self.assertSetEqual(set(["rhp1", "rhp2_named"]),
                            set(registry.list_ranged_hparams()))

Exemple #2

Fichier : registry_test.py Projet : qixiuai/tensor2tensor

  def testListHparams(self):

    @registry.register_hparams
    def hp1():
      pass

    @registry.register_hparams("hp2_named")
    def hp2():
      pass

    @registry.register_ranged_hparams
    def rhp1(_):
      pass

    @registry.register_ranged_hparams("rhp2_named")
    def rhp2(_):
      pass

    self.assertSetEqual(set(["hp1", "hp2_named"]), set(registry.list_hparams()))
    self.assertSetEqual(
        set(["rhp1", "rhp2_named"]), set(registry.list_ranged_hparams()))