업로드 시 exception 메세지 처리, 에폭 10 이상으로 실행되게 수정
This commit is contained in:
@@ -198,7 +198,10 @@ public class ModelTrainMngCoreService {
|
||||
ModelConfigEntity entity = new ModelConfigEntity();
|
||||
modelMasterEntity.setId(modelId);
|
||||
entity.setModel(modelMasterEntity);
|
||||
entity.setEpochCount(req.getEpochCnt());
|
||||
entity.setEpochCount(
|
||||
req.getEpochCnt() < 10
|
||||
? 10
|
||||
: req.getEpochCnt()); // 에폭이 10 이하이면 10으로 고정하기. 10 이상 에폭으로 해야 best 에폭 파일이 생성되어 내려옴
|
||||
entity.setTrainPercent(req.getTrainingCnt());
|
||||
entity.setValidationPercent(req.getValidationCnt());
|
||||
entity.setTestPercent(req.getTestCnt());
|
||||
|
||||
Reference in New Issue
Block a user