하이퍼 파라미터 사용회수 카운트 기능 추가 및 조회 수정
This commit is contained in:
@@ -34,7 +34,7 @@ public class HyperParamCoreService {
|
||||
|
||||
ModelHyperParamEntity entity = new ModelHyperParamEntity();
|
||||
entity.setHyperVer(firstVersion);
|
||||
|
||||
entity.setTotalUseCnt(entity.getTotalUseCnt() == null ? 1 : entity.getTotalUseCnt() + 1);
|
||||
applyHyperParam(entity, createReq);
|
||||
|
||||
// user
|
||||
@@ -172,7 +172,7 @@ public class HyperParamCoreService {
|
||||
} else {
|
||||
entity.setCropSize("256,256");
|
||||
}
|
||||
// entity.setCropSize(src.getCropSize());
|
||||
entity.setCropSize(src.getCropSize());
|
||||
|
||||
// Important
|
||||
entity.setModelType(model); // 20250212 modeltype추가
|
||||
|
||||
@@ -104,6 +104,9 @@ public class ModelTrainMngCoreService {
|
||||
if (hyperParamEntity == null || hyperParamEntity.getHyperVer() == null) {
|
||||
throw new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND);
|
||||
}
|
||||
// 하이퍼 파라미터 사용회수 업데이트
|
||||
hyperParamEntity.setTotalUseCnt(
|
||||
hyperParamEntity.getTotalUseCnt() == null ? 1 : hyperParamEntity.getTotalUseCnt() + 1);
|
||||
|
||||
String modelVer =
|
||||
String.join(
|
||||
|
||||
@@ -310,6 +310,9 @@ public class ModelHyperParamEntity {
|
||||
@Column(name = "default_param")
|
||||
private Boolean isDefault = false;
|
||||
|
||||
@Column(name = "total_use_cnt")
|
||||
private Integer totalUseCnt = 0;
|
||||
|
||||
public HyperParamDto.Basic toDto() {
|
||||
return new HyperParamDto.Basic(
|
||||
this.modelType,
|
||||
|
||||
@@ -296,14 +296,8 @@ public class TrainJobService {
|
||||
e);
|
||||
|
||||
// 런타임 예외로 래핑하되, 메시지에 핵심 정보 포함
|
||||
throw new IllegalStateException(
|
||||
"tmp dataset build failed: modelUuid="
|
||||
+ modelUuid
|
||||
+ ", modelId="
|
||||
+ modelId
|
||||
+ ", tmpRaw="
|
||||
+ raw,
|
||||
e);
|
||||
throw new CustomApiException(
|
||||
"INTERNAL_SERVER_ERROR", HttpStatus.INTERNAL_SERVER_ERROR, "임시 데이터셋 생성에 실패했습니다.");
|
||||
}
|
||||
return modelUuid;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user