Clarify which model is being initialized.
PiperOrigin-RevId: 239549766
This commit is contained in:
parent
ddf3966936
commit
0965a15b82
@ -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;
|
||||
|
Loading…
Reference in New Issue
Block a user