ShengZhaoDUT

如何export模型

9

我用如下代码导出DeepFM模型，但是发现freeze_model_dir/variables为空，请问应该如何导出DeepFM的pb模型？ `def freeze_model(self): freeze_model_dir = "freeze_model_dir" save_dir = 'checkpoints/' save_path = os.path.join(save_dir, 'best_validation') start_time = `time()` print(tf.trainable_variables()) print("freeze model...") SIGNATURE_NAME = "serving_default" builder = tf.saved_model.builder.SavedModelBuilder(freeze_model_dir) inputs = {'feat_index': tf.saved_model.utils.build_tensor_info(self.feat_index), 'feat_value':...