Skip to content

Commit 118c97f

Browse files
committed
add clear_param_diffs to the python net interface
1 parent f28f5ae commit 118c97f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/caffe/_caffe.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -271,6 +271,7 @@ BOOST_PYTHON_MODULE(_caffe) {
271271
.def("_forward", &Net<Dtype>::ForwardFromTo)
272272
.def("_backward", &Net<Dtype>::BackwardFromTo)
273273
.def("reshape", &Net<Dtype>::Reshape)
274+
.def("clear_param_diffs", &Net<Dtype>::ClearParamDiffs)
274275
// The cast is to select a particular overload.
275276
.def("copy_from", static_cast<void (Net<Dtype>::*)(const string)>(
276277
&Net<Dtype>::CopyTrainedLayersFrom))

0 commit comments

Comments
 (0)