Browse Source

实验模型手动导出

dev-lhz
chenzhihang 1 year ago
parent
commit
751b85b3ee
1 changed files with 5 additions and 2 deletions
  1. +5
    -2
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java

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

@@ -11,6 +11,7 @@ import com.ruoyi.platform.domain.AssetIcon;
import com.ruoyi.platform.domain.ModelDependency1;
import com.ruoyi.platform.domain.Models;
import com.ruoyi.platform.domain.ModelsVersion;
import com.ruoyi.platform.domain.dependencydomain.TrainTaskDepency;
import com.ruoyi.platform.mapper.ModelDependency1Dao;
import com.ruoyi.platform.mapper.ModelsDao;
import com.ruoyi.platform.mapper.ModelsVersionDao;
@@ -727,8 +728,10 @@ public class ModelsServiceImpl implements ModelsService {
String root_path = sourcePath.split("/")[1];
minioUtil.downloadFiles(bucketName, root_path, modelPath);

BeansUtils.copyPropertiesIgnoreNull(modelsVo, modelMetaVo);
buildingModel = modelDependency1Dao.queryByTrainTask(JSON.toJSONString(modelMetaVo));
HashMap<String, TrainTaskDepency> map = new HashMap<>();
map.put("trainTask", modelsVo.getTrainTask());
String t1 = JSON.toJSONString(map);
buildingModel = modelDependency1Dao.queryByTrainTask(t1);
} else {
dvcUtils.moveFiles(sourcePath, modelPath);



Loading…
Cancel
Save