Как я могу экспортировать модель с несколькими разделами данных - PullRequest
Новая
       11

Как я могу экспортировать модель с несколькими разделами данных

0 голосов
/ 28 марта 2019

Я хочу экспортировать модель из контрольной точки с помощью оценщика, а данные переменных хранятся в одном файле.Я хочу, чтобы переменные данные сохранялись в нескольких файлах, чтобы я мог быстрее их разархивировать.

Я попытался запустить несколько рабочих для экспорта, но это не работает.

Вот некоторые из моихкоды: 

  if self.job_name == "worker":
            if FLAGS.ckpt != '' and int(FLAGS.ckpt) > 0:
                estimator.export_savedmodel(self.export_model_path,
                                            receiver_fn_raw(self.params['dense_num'], self.params['feature_conf'],
                                                            self.params['list_num']),
                                            checkpoint_path=self.ckpt_dir)
            else:
                estimator.export_savedmodel(self.export_model_path,
                                            receiver_fn_raw(self.params['dense_num'], self.params['feature_conf'],
                                                            self.params['list_num']))
...