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