mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!30743 Fix an issue of federal differential privacy training. 

Merge pull request !30743 from jxlang910/r1.6
This commit is contained in:
i-robot 2022-03-02 12:25:10 +00:00 committed by Gitee
parent 4fb2ff5563 3fdf3f15b5
commit f1c1acd681
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/java/java/fl_client/src/main/java/com/mindspore/flclient/FLLiteClient.java
View File

@ -48,6 +48,7 @@ import java.util.List;
import java.util.Map;
import java.util.TreeMap;
import java.util.logging.Logger;
import java.util.ArrayList;
/**
* Defining the general process of federated learning tasks.
@ -490,7 +491,8 @@ public class FLLiteClient {
private float calWeightUpdateNorm(Map<String, float[]> originalData, Map<String, float[]> newData) {
float updateL2Norm = 0f;
for (String key : originalData.keySet()) {
ArrayList<String> featureList = secureProtocol.getUpdateFeatureName();
for (String key : featureList) {
float[] data = originalData.get(key);
float[] dataAfterUpdate = newData.get(key);
for (int j = 0; j < data.length; j++) {