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

fix fl client log print

This commit is contained in:
zhengjun10 2022-02-11 09:39:31 +08:00
parent 0823adbc5f
commit 2aa49034b4
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/lite/java/java/fl_client/src/main/java/com/mindspore/flclient/model/Client.java
View File

@ -239,7 +239,7 @@ public abstract class Client {
}
}
long endTime = System.currentTimeMillis();
logger.info(Common.addTag("total train time:" + (endTime - startTime) + "ms"));
logger.info(Common.addTag("total run time:" + (endTime - startTime) + "ms"));
return Status.SUCCESS;
}
@ -277,7 +277,8 @@ public abstract class Client {
}
trainSession = LiteSession.createSession(msConfig);
if (trainSession == null) {
logger.severe(Common.addTag("init session failed,please check model path:" + modelPath));
logger.severe(Common.addTag("init session failed,please check model :" + modelPath + " is valid or " +
"disk space is enough"));
msConfig.free();
model.free();
return Optional.empty();
@ -295,7 +296,8 @@ public abstract class Client {
} else {
trainSession = TrainSession.createTrainSession(modelPath, msConfig, false);
if (trainSession == null) {
logger.severe(Common.addTag("init session failed,please check model path:" + modelPath));
logger.severe(Common.addTag("init session failed,please check model :" + modelPath + " is valid or " +
"disk space is enough"));
return Optional.empty();
}
return Optional.of(trainSession);