Browse Source

Fix API Issue for SuppressCtrl

tags/v1.8.0
shu-kun-zhang 3 years ago
parent
commit
a59f1b39d1
1 changed files with 6 additions and 1 deletions
  1. +6
    -1
      mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py

+ 6
- 1
mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py View File

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


Loading…
Cancel
Save