allen 11 сар өмнө
parent
commit
06752271fa

+ 1 - 1
taais-modules/taais-biz/src/main/java/com/taais/biz/mapper/AlgorithmSubtaskMapper.java

@@ -19,5 +19,5 @@ public interface AlgorithmSubtaskMapper extends BaseMapper<AlgorithmSubtask> {
 
     List<AlgorithmSubtaskVo> getSubtaskByTaskId(Long taskId);
 
-    String getTrainModelPath(Long id, Long algorithmId);
+    String getTrainModelPath(Long id);
 }

+ 2 - 1
taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/AlgorithmSubtaskServiceImpl.java

@@ -242,7 +242,7 @@ public class AlgorithmSubtaskServiceImpl extends BaseServiceImpl<AlgorithmSubtas
                     SysOssVo modelOss = ossService.getById(Long.valueOf(model.getModelAddress()));
                     otherParams.put("pretrained_model", BizConstant.DOCKER_BASE_PATH + StringUtils.substringAfter(modelOss.getFileName(), Constants.RESOURCE_PREFIX));
                     // 找到训练的模型地址
-                    String trainModelPath = mapper.getTrainModelPath(bizProcessVo.getId(), bizProcessVo.getAlgorithmId());
+                    String trainModelPath = mapper.getTrainModelPath(bizProcessVo.getId());
                     if (StringUtils.isNotEmpty(trainModelPath)){
                         otherParams.put("weight_path", BizConstant.DOCKER_BASE_PATH + trainModelPath + BizConstant.DOCKER_PT_PATH);
                     }
@@ -261,6 +261,7 @@ public class AlgorithmSubtaskServiceImpl extends BaseServiceImpl<AlgorithmSubtas
             if (StringUtils.isEmpty(url)) {
                 errorMsg.add("url是空!!!");
             } else {
+                log.info("http post url:{},body:{}", url,algorithmRequestDto);
                 WebClient webClient = webClientBuilder.build();
                 Mono<String> response = webClient.post()
                     .uri(url)