如何导出具有多个数据分区的模型

Question

我想使用估算器从检查点导出模型，并将变量数据存储在一个文件中。我希望将变量数据存储在多个文件中，以便可以更快地进行分配。

我尝试启动多名工人进行出口，但没有用。

这是我的一些代码：

  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']))