diff options
| author | Liang Xiong <lxiong@fb.com> | 2019-04-09 09:25:03 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-04-09 09:35:43 -0700 |
| commit | b1bea0b7331fdac736a17918579fa1fd24e183a7 (patch) | |
| tree | e9660d801f3a94e2806bccde2c402f6a7ad51520 /caffe2 | |
| parent | 89145e602b91064a87c7f57bfe58364c145f69bd (diff) | |
| download | pytorch-b1bea0b7331fdac736a17918579fa1fd24e183a7.tar.gz pytorch-b1bea0b7331fdac736a17918579fa1fd24e183a7.tar.bz2 pytorch-b1bea0b7331fdac736a17918579fa1fd24e183a7.zip | |
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
Diffstat (limited to 'caffe2')
| -rw-r--r-- | caffe2/operators/load_save_op.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/caffe2/operators/load_save_op.h b/caffe2/operators/load_save_op.h index 565761199d..a901d8e827 100644 --- a/caffe2/operators/load_save_op.h +++ b/caffe2/operators/load_save_op.h @@ -485,6 +485,8 @@ class SaveOp final : public Operator<Context> { }; const vector<const Blob*>& 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_); } |