Exemplos de get_policy_model_file_from_epoch em Python

Exemplo n.º 1

Arquivo: ppo_test.py Projeto: levskaya/tensor2tensor

 def write_policy_model_file(epoch):
     with gfile.GFile(
             ppo.get_policy_model_file_from_epoch(output_dir, epoch),
             "w") as f:
         f.write("some data")

Exemplo n.º 2

Arquivo: ppo_test.py Projeto: levskaya/tensor2tensor

 def test_get_policy_model_file_from_epoch(self):
     self.assertEqual("/tmp/model-000000.pkl",
                      ppo.get_policy_model_file_from_epoch("/tmp", 0))
     self.assertEqual("/tmp/model-123456.pkl",
                      ppo.get_policy_model_file_from_epoch("/tmp", 123456))

Exemplo n.º 3

Arquivo: async_trajectory_collector_lib_test.py Projeto: zhouyonglong/tensor2tensor

 def write_policy_model_file(epoch):
     fname = ppo.get_policy_model_file_from_epoch(output_dir, epoch)
     with gfile.GFile(fname, "w") as f:
         f.write("some data")
     return fname