Browse Source

优化主动学习

pull/225/head
chenzhihang 8 months ago
parent
commit
7b51ed5bc6
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnInsServiceImpl.java

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

@@ -293,6 +293,8 @@ public class ActiveLearnInsServiceImpl implements ActiveLearnInsService {

public void getTrialList(ActiveLearnIns ins) throws IOException {
String directoryPath = ins.getResultPath();
Map<String, Object> stringObjectMap = JsonUtils.jsonToMap(ins.getParam());
String s = (String) stringObjectMap.get("framework_type");
switch ((String) JsonUtils.jsonToMap(ins.getParam()).get("framework_type")) {
case Constant.AL_PYTORCH: {
ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model_weights.pth");


Loading…
Cancel
Save