MVPTeam/kamco-train-api
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat/training_260202 #3

Merged
teddy merged 3 commits from feat/training_260202 into develop 2026-02-02 19:32:22 +09:00
Conversation 0 Commits 3 Files Changed 9 +174 -192
3 changed files with 174 additions and 192 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit d168e9712e - Show all commits

1
src/main/java/com/kamco/cd/training/model/dto/HyperParamDto.java
View File

@@ -138,7 +138,6 @@ public class HyperParamDto {
this.hueDelta = entity.getHueDelta(); this.hueDelta = entity.getHueDelta();
// Legacy // Legacy
this.dropoutRatio = entity.getDropoutRatio();
this.cnnFilterCnt = entity.getCnnFilterCnt(); this.cnnFilterCnt = entity.getCnnFilterCnt();
// Common // Common

5
src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java
View File

@@ -77,7 +77,6 @@ public class HyperParamCoreService {
.saturationRange(entity.getSaturationRange()) .saturationRange(entity.getSaturationRange())
.hueDelta(entity.getHueDelta()) .hueDelta(entity.getHueDelta())
// Legacy // Legacy
.dropoutRatio(entity.getDropoutRatio())
.cnnFilterCnt(entity.getCnnFilterCnt()) .cnnFilterCnt(entity.getCnnFilterCnt())
// Common // Common
.memo(entity.getMemo()) .memo(entity.getMemo())
@@ -247,10 +246,6 @@ public class HyperParamCoreService {
createReq.getHueDelta() != null ? createReq.getHueDelta() : baseEntity.getHueDelta()); createReq.getHueDelta() != null ? createReq.getHueDelta() : baseEntity.getHueDelta());
// Legacy // Legacy
entity.setDropoutRatio(
createReq.getDropoutRatio() != null
? createReq.getDropoutRatio()
: baseEntity.getDropoutRatio());
entity.setCnnFilterCnt( entity.setCnnFilterCnt(
createReq.getCnnFilterCnt() != null createReq.getCnnFilterCnt() != null
? createReq.getCnnFilterCnt() ? createReq.getCnnFilterCnt()

360
src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java
View File

@@ -2,231 +2,219 @@ package com.kamco.cd.training.postgres.entity;
import jakarta.persistence.Column; import jakarta.persistence.Column;
import jakarta.persistence.Entity; import jakarta.persistence.Entity;
import jakarta.persistence.GeneratedValue;
import jakarta.persistence.GenerationType;
import jakarta.persistence.Id; import jakarta.persistence.Id;
import jakarta.persistence.Table; import jakarta.persistence.Table;
import jakarta.validation.constraints.NotNull; import jakarta.validation.constraints.NotNull;
import jakarta.validation.constraints.Size; import jakarta.validation.constraints.Size;
import java.time.ZonedDateTime; import java.time.ZonedDateTime;
import lombok.AllArgsConstructor; import java.util.UUID;
import lombok.Getter; import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter; import lombok.Setter;
import org.hibernate.annotations.ColumnDefault; import org.hibernate.annotations.ColumnDefault;
@Getter @Getter
@Setter @Setter
@Entity @Entity
@NoArgsConstructor @Table(name = "tb_model_hyper_params")
@AllArgsConstructor
@Table(name = "tb_model_hyper_param")
public class ModelHyperParamEntity { public class ModelHyperParamEntity {
@Id @Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
@Column(name = "hyper_param_id", nullable = false)
private Long id;
@NotNull @NotNull
@ColumnDefault("gen_random_uuid()")
@Column(name = "uuid", nullable = false)
private UUID uuid = UUID.randomUUID();
@Size(max = 50) @Size(max = 50)
@NotNull
@Column(name = "hyper_ver", nullable = false, length = 50) @Column(name = "hyper_ver", nullable = false, length = 50)
private String hyperVer; private String hyperVer;
// ==================== Important Parameters ==================== @Size(max = 20)
@NotNull
@Column(name = "backbone", nullable = false, length = 20)
private String backbone;
@Size(max = 15)
@NotNull
@Column(name = "input_size", nullable = false, length = 15)
private String inputSize;
@Size(max = 15)
@NotNull
@Column(name = "crop_size", nullable = false, length = 15)
private String cropSize;
@NotNull
@Column(name = "epoch_cnt", nullable = false)
private Integer epochCnt;
@NotNull
@Column(name = "batch_size", nullable = false)
private Integer batchSize;
@NotNull
@Column(name = "drop_path_rate", nullable = false)
private Double dropPathRate;
@NotNull
@Column(name = "frozen_stages", nullable = false)
private Integer frozenStages;
@Size(max = 20) @Size(max = 20)
@ColumnDefault("'large'") @NotNull
@Column(name = "backbone", length = 20) @Column(name = "neck_policy", nullable = false, length = 20)
private String backbone = "large"; private String neckPolicy;
@Size(max = 20)
@ColumnDefault("'256,256'")
@Column(name = "input_size", length = 20)
private String inputSize = "256,256";
@Size(max = 20)
@ColumnDefault("'256,256'")
@Column(name = "crop_size", length = 20)
private String cropSize = "256,256";
@ColumnDefault("200")
@Column(name = "epoch_cnt")
private Integer epochCnt = 200;
@ColumnDefault("16")
@Column(name = "batch_size")
private Integer batchSize = 16;
// ==================== Model Architecture ====================
@ColumnDefault("0.3")
@Column(name = "drop_path_rate")
private Double dropPathRate = 0.3;
@ColumnDefault("-1")
@Column(name = "frozen_stages")
private Integer frozenStages = -1;
@Size(max = 20)
@ColumnDefault("'abs_diff'")
@Column(name = "neck_policy", length = 20)
private String neckPolicy = "abs_diff";
@Size(max = 255)
@ColumnDefault("'512,256,128,64'")
@Column(name = "decoder_channels", length = 255)
private String decoderChannels = "512,256,128,64";
@Size(max = 500)
@Column(name = "class_weight", length = 500)
private String classWeight;
@Column(name = "num_layers")
private Integer numLayers;
// ==================== Loss & Optimization ====================
@ColumnDefault("0.00006")
@Column(name = "learning_rate")
private Double learningRate = 0.00006;
@ColumnDefault("0.05")
@Column(name = "weight_decay")
private Double weightDecay = 0.05;
@ColumnDefault("0.9")
@Column(name = "layer_decay_rate")
private Double layerDecayRate = 0.9;
@ColumnDefault("true")
@Column(name = "ddp_find_unused_params")
private Boolean ddpFindUnusedParams = true;
@ColumnDefault("255")
@Column(name = "ignore_index")
private Integer ignoreIndex = 255;
// ==================== Data ====================
@ColumnDefault("16")
@Column(name = "train_num_workers")
private Integer trainNumWorkers = 16;
@ColumnDefault("8")
@Column(name = "val_num_workers")
private Integer valNumWorkers = 8;
@ColumnDefault("8")
@Column(name = "test_num_workers")
private Integer testNumWorkers = 8;
@ColumnDefault("true")
@Column(name = "train_shuffle")
private Boolean trainShuffle = true;
@ColumnDefault("true")
@Column(name = "train_persistent")
private Boolean trainPersistent = true;
@ColumnDefault("true")
@Column(name = "val_persistent")
private Boolean valPersistent = true;
// ==================== Evaluation ====================
@Size(max = 255)
@ColumnDefault("'mFscore,mIoU'")
@Column(name = "metrics", length = 255)
private String metrics = "mFscore,mIoU";
@Size(max = 50) @Size(max = 50)
@ColumnDefault("'changed_fscore'") @NotNull
@Column(name = "save_best", length = 50) @Column(name = "decoder_channels", nullable = false, length = 50)
private String saveBest = "changed_fscore"; private String decoderChannels;
@Size(max = 20) @Size(max = 50)
@ColumnDefault("'greater'") @NotNull
@Column(name = "save_best_rule", length = 20) @Column(name = "class_weight", nullable = false, length = 50)
private String saveBestRule = "greater"; private String classWeight;
@ColumnDefault("10") @NotNull
@Column(name = "val_interval") @Column(name = "num_layers", nullable = false)
private Integer valInterval = 10; private Integer numLayers;
@ColumnDefault("400") @NotNull
@Column(name = "log_interval") @Column(name = "learning_rate", nullable = false)
private Integer logInterval = 400; private Double learningRate;
@ColumnDefault("1") @NotNull
@Column(name = "vis_interval") @Column(name = "weight_decay", nullable = false)
private Integer visInterval = 1; private Double weightDecay;
// ==================== Hardware ==================== @NotNull
@Column(name = "layer_decay_rate", nullable = false)
private Double layerDecayRate;
@ColumnDefault("4") @NotNull
@Column(name = "gpu_cnt") @Column(name = "ddp_find_unused_params", nullable = false)
private Integer gpuCnt = 4; private Boolean ddpFindUnusedParams = false;
@NotNull
@Column(name = "ignore_index", nullable = false)
private Integer ignoreIndex;
@NotNull
@Column(name = "train_num_workers", nullable = false)
private Integer trainNumWorkers;
@NotNull
@Column(name = "val_num_workers", nullable = false)
private Integer valNumWorkers;
@NotNull
@Column(name = "test_num_workers", nullable = false)
private Integer testNumWorkers;
@NotNull
@Column(name = "train_shuffle", nullable = false)
private Boolean trainShuffle = false;
@NotNull
@Column(name = "train_persistent", nullable = false)
private Boolean trainPersistent = false;
@NotNull
@Column(name = "val_persistent", nullable = false)
private Boolean valPersistent = false;
@Size(max = 100) @Size(max = 100)
@ColumnDefault("'0,1,2,3'") @NotNull
@Column(name = "gpu_ids", length = 100) @Column(name = "metrics", nullable = false, length = 100)
private String gpuIds = "0,1,2,3"; private String metrics;
@Size(max = 30)
@NotNull
@Column(name = "save_best", nullable = false, length = 30)
private String saveBest;
@Size(max = 10)
@NotNull
@Column(name = "save_best_rule", nullable = false, length = 10)
private String saveBestRule;
@NotNull
@Column(name = "val_interval", nullable = false)
private Integer valInterval;
@NotNull
@Column(name = "log_interval", nullable = false)
private Integer logInterval;
@NotNull
@Column(name = "vis_interval", nullable = false)
private Integer visInterval;
@NotNull
@Column(name = "rot_prob", nullable = false)
private Double rotProb;
@NotNull
@Column(name = "flip_prob", nullable = false)
private Double flipProb;
@Size(max = 20)
@NotNull
@Column(name = "rot_degree", nullable = false, length = 20)
private String rotDegree;
@NotNull
@Column(name = "exchange_prob", nullable = false)
private Double exchangeProb;
@NotNull
@Column(name = "brightness_delta", nullable = false)
private Integer brightnessDelta;
@Size(max = 20)
@NotNull
@Column(name = "contrast_range", nullable = false, length = 20)
private String contrastRange;
@Size(max = 20)
@NotNull
@Column(name = "saturation_range", nullable = false, length = 20)
private String saturationRange;
@NotNull
@Column(name = "hue_delta", nullable = false)
private Integer hueDelta;
@NotNull
@Column(name = "gpu_cnt", nullable = false)
private Integer gpuCnt;
@Size(max = 50)
@Column(name = "gpu_ids", length = 50)
private String gpuIds;
@ColumnDefault("1122")
@Column(name = "master_port") @Column(name = "master_port")
private Integer masterPort = 1122; private Integer masterPort;
// ==================== Augmentation ====================
@ColumnDefault("0.5")
@Column(name = "rot_prob")
private Double rotProb = 0.5;
@ColumnDefault("0.5")
@Column(name = "flip_prob")
private Double flipProb = 0.5;
@Size(max = 20)
@ColumnDefault("'-20,20'")
@Column(name = "rot_degree", length = 20)
private String rotDegree = "-20,20";
@ColumnDefault("0.5")
@Column(name = "exchange_prob")
private Double exchangeProb = 0.5;
@ColumnDefault("10")
@Column(name = "brightness_delta")
private Integer brightnessDelta = 10;
@Size(max = 20)
@ColumnDefault("'0.8,1.2'")
@Column(name = "contrast_range", length = 20)
private String contrastRange = "0.8,1.2";
@Size(max = 20)
@ColumnDefault("'0.8,1.2'")
@Column(name = "saturation_range", length = 20)
private String saturationRange = "0.8,1.2";
@ColumnDefault("10")
@Column(name = "hue_delta")
private Integer hueDelta = 10;
// ==================== Legacy (deprecated) ====================
@Column(name = "cnn_filter_cnt")
private Integer cnnFilterCnt;
@Column(name = "dropout_ratio")
private Double dropoutRatio;
// ==================== Common ====================
@Column(name = "memo", length = Integer.MAX_VALUE) @Column(name = "memo", length = Integer.MAX_VALUE)
private String memo; private String memo;
@Size(max = 255) @NotNull
@ColumnDefault("'N'") @ColumnDefault("'N'")
@Column(name = "del_yn", length = 255) @Column(name = "del_yn", nullable = false, length = Integer.MAX_VALUE)
private String delYn = "N"; private String delYn;
@ColumnDefault("now()") @NotNull
@Column(name = "created_dttm") @ColumnDefault("CURRENT_TIMESTAMP")
@Column(name = "created_dttm", nullable = false)
private ZonedDateTime createdDttm; private ZonedDateTime createdDttm;
@Column(name = "cnn_filter_cnt")
private Integer cnnFilterCnt;
} }