데이터셋 등록 추가

src/main/java/com/kamco/cd/training/common/dto/HyperParam.java

@@ -0,0 +1,154 @@
+package com.kamco.cd.training.common.dto;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+@Getter
+@Setter
+@AllArgsConstructor
+@NoArgsConstructor
+public class HyperParam {
+  // -------------------------
+  // Important
+  // -------------------------
+  @Schema(description = "백본 네트워크", example = "large")
+  private String backbone; // backbone
+
+  @Schema(description = "입력 이미지 크기(H,W)", example = "512,512")
+  private String inputSize; // input_size
+
+  @Schema(description = "크롭 크기(H,W 또는 단일값)", example = "256,256")
+  private String cropSize; // crop_size
+
+  @Schema(description = "배치 크기(Per GPU)", example = "16")
+  private Integer batchSize; // batch_size
+
+  // -------------------------
+  // Data
+  // -------------------------
+  @Schema(description = "Train dataloader workers", example = "16")
+  private Integer trainNumWorkers; // train_num_workers
+
+  @Schema(description = "Val dataloader workers", example = "8")
+  private Integer valNumWorkers; // val_num_workers
+
+  @Schema(description = "Test dataloader workers", example = "8")
+  private Integer testNumWorkers; // test_num_workers
+
+  @Schema(description = "Train shuffle 여부", example = "true")
+  private Boolean trainShuffle; // train_shuffle
+
+  @Schema(description = "Train persistent workers 여부", example = "true")
+  private Boolean trainPersistent; // train_persistent
+
+  @Schema(description = "Val persistent workers 여부", example = "true")
+  private Boolean valPersistent; // val_persistent
+
+  // -------------------------
+  // Model Architecture
+  // -------------------------
+  @Schema(description = "Drop Path 비율", example = "0.3")
+  private Double dropPathRate; // drop_path_rate
+
+  @Schema(description = "Freeze 단계(-1:None)", example = "-1")
+  private Integer frozenStages; // frozen_stages
+
+  @Schema(description = "Neck 결합 정책", example = "abs_diff")
+  private String neckPolicy; // neck_policy
+
+  @Schema(description = "디코더 채널 구성", example = "512,256,128,64")
+  private String decoderChannels; // decoder_channels
+
+  @Schema(description = "클래스별 가중치", example = "1,10")
+  private String classWeight; // class_weight
+
+  // -------------------------
+  // Loss & Optimization
+  // -------------------------
+  @Schema(description = "학습률", example = "0.00006")
+  private Double learningRate; // learning_rate
+
+  @Schema(description = "Weight Decay", example = "0.05")
+  private Double weightDecay; // weight_decay
+
+  @Schema(description = "Layer Decay Rate", example = "0.9")
+  private Double layerDecayRate; // layer_decay_rate
+
+  @Schema(description = "DDP unused params 탐색 여부", example = "true")
+  private Boolean ddpFindUnusedParams; // ddp_find_unused_params
+
+  @Schema(description = "Loss 계산 제외 인덱스", example = "255")
+  private Integer ignoreIndex; // ignore_index
+
+  @Schema(description = "레이어 깊이", example = "24")
+  private Integer numLayers; // num_layers
+
+  // -------------------------
+  // Evaluation
+  // -------------------------
+  @Schema(description = "평가 지표 목록", example = "mFscore,mIoU")
+  private String metrics; // metrics
+
+  @Schema(description = "Best 모델 선정 기준 지표", example = "changed_fscore")
+  private String saveBest; // save_best
+
+  @Schema(description = "Best 모델 선정 규칙", example = "less")
+  private String saveBestRule; // save_best_rule
+
+  @Schema(description = "검증 수행 주기(Epoch)", example = "10")
+  private Integer valInterval; // val_interval
+
+  @Schema(description = "로그 기록 주기(Iteration)", example = "400")
+  private Integer logInterval; // log_interval
+
+  @Schema(description = "시각화 저장 주기(Epoch)", example = "1")
+  private Integer visInterval; // vis_interval
+
+  // -------------------------
+  // Augmentation
+  // -------------------------
+  @Schema(description = "회전 적용 확률", example = "0.5")
+  private Double rotProb; // rot_prob
+
+  @Schema(description = "회전 각도 범위(Min,Max)", example = "-20,20")
+  private String rotDegree; // rot_degree
+
+  @Schema(description = "반전 적용 확률", example = "0.5")
+  private Double flipProb; // flip_prob
+
+  @Schema(description = "채널 교환 확률", example = "0.5")
+  private Double exchangeProb; // exchange_prob
+
+  @Schema(description = "밝기 변화량", example = "10")
+  private Integer brightnessDelta; // brightness_delta
+
+  @Schema(description = "대비 범위(Min,Max)", example = "0.8,1.2")
+  private String contrastRange; // contrast_range
+
+  @Schema(description = "채도 범위(Min,Max)", example = "0.8,1.2")
+  private String saturationRange; // saturation_range
+
+  @Schema(description = "색조 변화량", example = "10")
+  private Integer hueDelta; // hue_delta
+
+  // -------------------------
+  // Hardware
+  // -------------------------
+  @Schema(description = "사용 GPU 개수", example = "4")
+  private Integer gpuCnt; // gpu_cnt
+
+  @Schema(description = "사용 GPU ID 목록", example = "0,1,2,3")
+  private String gpuIds; // gpu_ids
+
+  @Schema(description = "분산학습 마스터 포트", example = "1122")
+  private Integer masterPort; // master_port
+
+  // -------------------------
+  // Memo
+  // -------------------------
+  @Schema(description = "메모", example = "하이퍼파라미터 신규등록")
+  private String memo; // memo
+}

src/main/java/com/kamco/cd/training/common/enums/ModelType.java

@@ -0,0 +1,27 @@
+package com.kamco.cd.training.common.enums;
+
+import com.kamco.cd.training.common.utils.enums.CodeExpose;
+import com.kamco.cd.training.common.utils.enums.EnumType;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+@CodeExpose
+@Getter
+@AllArgsConstructor
+public enum ModelType implements EnumType {
+  M1("M1"),
+  M2("M2"),
+  M3("M3");
+
+  private String desc;
+
+  @Override
+  public String getId() {
+    return name();
+  }
+
+  @Override
+  public String getText() {
+    return desc;
+  }
+}