전이학습 상세 수정
This commit is contained in:
@@ -93,6 +93,29 @@ public class ModelTrainDetailDto {
|
||||
private Integer batchSize;
|
||||
}
|
||||
|
||||
@Schema(name = "모델학습관리 전이 하이파라미터", description = "모델학습관리 전이 하이파라미터")
|
||||
@Getter
|
||||
@Setter
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
@Builder
|
||||
public static class TransferHyperSummary {
|
||||
private UUID uuid;
|
||||
private Long hyperParamId;
|
||||
private String hyperVer;
|
||||
private String backbone;
|
||||
private String inputSize;
|
||||
private String cropSize;
|
||||
private Integer batchSize;
|
||||
private UUID beforeUuid;
|
||||
private Long beforeHyperParamId;
|
||||
private String beforeHyperVer;
|
||||
private String beforeBackbone;
|
||||
private String beforeInputSize;
|
||||
private String beforeCropSize;
|
||||
private Integer beforeBatchSize;
|
||||
}
|
||||
|
||||
@Schema(name = "선택한 데이터셋 목록", description = "선택한 데이터셋 목록")
|
||||
@Getter
|
||||
@Setter
|
||||
@@ -154,7 +177,7 @@ public class ModelTrainDetailDto {
|
||||
@AllArgsConstructor
|
||||
public static class TransferDetailDto {
|
||||
private ModelConfigDto.Basic etcConfig;
|
||||
private HyperSummary modelTrainHyper;
|
||||
private TransferHyperSummary modelTrainHyper;
|
||||
private List<SelectDataSet> modelTrainDataset;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -137,6 +137,9 @@ public class ModelTrainMngDto {
|
||||
@Schema(description = "학습타입 GENERAL(일반), TRANSFER(전이)", example = "GENERAL")
|
||||
private String trainType;
|
||||
|
||||
@Schema(description = "전이학습일때 선택한 모델 id")
|
||||
private Long beforeModelId;
|
||||
|
||||
@NotNull
|
||||
@Schema(
|
||||
description = "하이퍼 파라미터 선택 타입 OPTIMIZED(최적화 파라미터),EXISTING(기존 파라미터),NEW(신규 파라미터)",
|
||||
|
||||
Reference in New Issue
Block a user