From a59f1b39d1be7c904a3f92543d4c5fabb5a637e3 Mon Sep 17 00:00:00 2001 From: shu-kun-zhang Date: Sat, 26 Mar 2022 11:08:18 +0800 Subject: [PATCH] Fix API Issue for SuppressCtrl --- mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py b/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py index 7c15eb3..b12d0a2 100644 --- a/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py +++ b/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py @@ -676,7 +676,12 @@ class SuppressCtrl(Cell): return sparse, sparse_value_1, sparse_value_2 def calc_actual_sparse_for_fc1(self, networks): - """calculate actual sparse for full connection 1 layer""" + """ + Calculate actual sparse for full connection 1 layer + + Args: + networks (Cell): The training network. + """ return self.calc_actual_sparse_for_layer(networks, "fc1.weight") def calc_actual_sparse_for_layer(self, networks, layer_name):