Clarify which model is being initialized.

PiperOrigin-RevId: 239549766
This commit is contained in:
A. Unique TensorFlower 2019-03-21 00:43:53 -07:00 committed by TensorFlower Gardener
parent ddf3966936
commit 0965a15b82

View File

@ -148,7 +148,8 @@ Status RunInitOp(const RunOptions& run_options, const string& export_dir,
const std::vector<AssetFileDef>& asset_file_defs,
Session* session, const string& init_op_name) {
if (!init_op_name.empty()) {
LOG(INFO) << "Running initialization op on SavedModel bundle.";
LOG(INFO) << "Running initialization op on SavedModel bundle at path: "
<< export_dir;
std::vector<std::pair<string, Tensor>> inputs;
AddAssetsTensorsToInputs(export_dir, asset_file_defs, &inputs);
RunMetadata run_metadata;