OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Add missing .get_final_data_gradient() for repeat layer

This commit is contained in:
Davis King 2021-01-28 08:35:17 -05:00
parent 1516cf31c3
commit 9f6aefc0db
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dlib/dnn/core.h
View File

@ -1901,6 +1901,9 @@ namespace dlib
return details[0].get_gradient_input();
}
const tensor& get_final_data_gradient(
) const { return subnetwork.get_final_data_gradient(); }
const tensor& get_parameter_gradient(
) const { return details[0].get_parameter_gradient(); }