From b1bea0b7331fdac736a17918579fa1fd24e183a7 Mon Sep 17 00:00:00 2001 From: Liang Xiong Date: Tue, 9 Apr 2019 09:25:03 -0700 Subject: [PATCH] add logging to make the saving action visible (#19042) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19042 show the model saving step in the log. Reviewed By: kennyhorror Differential Revision: D14809385 fbshipit-source-id: c7a1e50ff92bb45b16b1c501d9325b304b07fbd3 --- caffe2/operators/load_save_op.h | 2 ++ 1 file changed, 2 insertions(+) diff --git a/caffe2/operators/load_save_op.h b/caffe2/operators/load_save_op.h index 5657611..a901d8e 100644 --- a/caffe2/operators/load_save_op.h +++ b/caffe2/operators/load_save_op.h @@ -485,6 +485,8 @@ class SaveOp final : public Operator { }; const vector& inputs = OperatorBase::Inputs(); + VLOG(0) << "Saving " << inputs.size() << " inputs to " << db_type_ << ": " + << full_db_name; for (int i = 0; i < inputs.size(); ++i) { SerializeBlob(*inputs[i], blob_names_[i], acceptor, chunk_size_); } -- 2.7.4