Browse Source

优化主动学习

pull/227/head
chenzhihang 10 months ago
parent
commit
04a9f8f125
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnInsServiceImpl.java

+ 1
- 2
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnInsServiceImpl.java View File

@@ -295,8 +295,7 @@ public class ActiveLearnInsServiceImpl implements ActiveLearnInsService {
String directoryPath = ins.getResultPath();
Map<String, Object> paramMap = JsonUtils.jsonToMap(ins.getParam());
Map<String, Object> dataMap = (Map<String, Object>) paramMap.get("data");
String s = (String) dataMap.get("framework_type");
switch ((String) JsonUtils.jsonToMap(ins.getParam()).get("framework_type")) {
switch ((String) dataMap.get("framework_type")) {
case Constant.AL_PYTORCH: {
ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model_weights.pth");
}


Loading…
Cancel
Save