Преглед на файлове

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

Eagle преди 7 месеца
родител
ревизия
083cea1d60
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
  1. 1 0
      taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/TargetIdentificationSubtaskServiceImpl.java

+ 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();