Esempi in Python per Decoder

Esempio n. 1

File: test_trainer_utils.py Progetto: wzhang1/lingvo

        def DecoderBuilds(self, model):
            tmpdir = os.path.join(FLAGS.test_tmpdir, model.__name__)
            tf.io.gfile.makedirs(tmpdir)

            model_params = model()
            cfg = model_params.Model()
            cfg.input = model_params.GetDatasetParams('Train')
            cfg.cluster.mode = 'sync'
            cfg.cluster.job = 'decoder'
            cfg.cluster.task = 0
            cfg.cluster.decoder.replicas = 1
            _ = trainer_lib.Decoder('train', cfg, '', tmpdir, tf_master='')

Esempio n. 2

File: trainer_test.py Progetto: yueyedeai/lingvo

 def _CreateDecoderDev(self, cfg):
     return trainer.Decoder('dev', cfg, FLAGS.model_task_name, FLAGS.logdir,
                            FLAGS.tf_master, self._trial)