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):