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='')
def _CreateDecoderDev(self, cfg): return trainer.Decoder('dev', cfg, FLAGS.model_task_name, FLAGS.logdir, FLAGS.tf_master, self._trial)