Эх сурвалжийг харах

[FIX] 同时提供weight_path和pretrained_model两个参数

Eagle 7 сар өмнө
parent
commit
083cea1d60

+ 1 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/TargetIdentificationSubtaskServiceImpl.java

@@ -244,6 +244,7 @@ public class TargetIdentificationSubtaskServiceImpl extends BaseServiceImpl<Targ
             String path = bo.getModelAddress().replace("/profile", BizConstant.DOCKER_BASE_PATH);
             algorithmRequestDto.getOtherParams().put("pretrained", true);
             algorithmRequestDto.getOtherParams().put("weight_path", path);
+            algorithmRequestDto.getOtherParams().put("pretrained_model", path);
         }
 
         String taskName = detail.getName();