추론 실행 추가
This commit is contained in:
@@ -0,0 +1,15 @@
|
||||
package com.kamco.cd.training.train.dto;
|
||||
|
||||
/** 학습 실행이 예약되었음을 알리는 이벤트 객체 */
|
||||
public class ModelTrainJobQueuedEvent {
|
||||
|
||||
private final Long jobId;
|
||||
|
||||
public ModelTrainJobQueuedEvent(Long jobId) {
|
||||
this.jobId = jobId;
|
||||
}
|
||||
|
||||
public Long getJobId() {
|
||||
return jobId;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,82 @@
|
||||
package com.kamco.cd.training.train.dto;
|
||||
|
||||
import lombok.AllArgsConstructor;
|
||||
import lombok.Getter;
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.Setter;
|
||||
|
||||
@Getter
|
||||
@Setter
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
public class TrainRunRequest {
|
||||
|
||||
// ========================
|
||||
// 기본
|
||||
// ========================
|
||||
private String datasetFolder;
|
||||
private String outputFolder;
|
||||
private String inputSize;
|
||||
private String cropSize;
|
||||
private Integer batchSize;
|
||||
private String gpuIds;
|
||||
private Integer gpus;
|
||||
private Double learningRate;
|
||||
private String backbone;
|
||||
private Integer epochs;
|
||||
|
||||
// ========================
|
||||
// Data
|
||||
// ========================
|
||||
private Integer trainNumWorkers;
|
||||
private Integer valNumWorkers;
|
||||
private Integer testNumWorkers;
|
||||
private Boolean trainShuffle;
|
||||
private Boolean trainPersistent;
|
||||
private Boolean valPersistent;
|
||||
|
||||
// ========================
|
||||
// Model Architecture
|
||||
// ========================
|
||||
private Double dropPathRate;
|
||||
private Integer frozenStages;
|
||||
private String neckPolicy;
|
||||
private String classWeight;
|
||||
private String decoderChannels;
|
||||
|
||||
// ========================
|
||||
// Loss & Optimization
|
||||
// ========================
|
||||
private Double weightDecay;
|
||||
private Double layerDecayRate;
|
||||
private Integer ignoreIndex;
|
||||
private Boolean ddpFindUnusedParams;
|
||||
private Integer numLayers;
|
||||
|
||||
// ========================
|
||||
// Evaluation
|
||||
// ========================
|
||||
private String metrics;
|
||||
private String saveBest;
|
||||
private String saveBestRule;
|
||||
private Integer valInterval;
|
||||
private Integer logInterval;
|
||||
private Integer visInterval;
|
||||
|
||||
// ========================
|
||||
// Augmentation
|
||||
// ========================
|
||||
private Double rotProb;
|
||||
private String rotDegree;
|
||||
private Double flipProb;
|
||||
private Double exchangeProb;
|
||||
private Integer brightnessDelta;
|
||||
private String contrastRange;
|
||||
private String saturationRange;
|
||||
private Integer hueDelta;
|
||||
|
||||
// ========================
|
||||
// 실행 타임아웃
|
||||
// ========================
|
||||
private Integer timeoutSeconds;
|
||||
}
|
||||
@@ -0,0 +1,20 @@
|
||||
package com.kamco.cd.training.train.dto;
|
||||
|
||||
import lombok.AllArgsConstructor;
|
||||
import lombok.Getter;
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.Setter;
|
||||
|
||||
/** 학습 실행 결과 반환 객체 */
|
||||
@Getter
|
||||
@Setter
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
public class TrainRunResult {
|
||||
|
||||
private String jobId;
|
||||
private String containerName;
|
||||
private int exitCode;
|
||||
private String status;
|
||||
private String logs;
|
||||
}
|
||||
Reference in New Issue
Block a user