fix(ocr): fix segmentation training for ketos 7 and low-memory hosts #234
@@ -162,11 +162,12 @@ public class OcrTrainingService {
|
|||||||
return Objects.requireNonNull(txTemplate.execute(status -> {
|
return Objects.requireNonNull(txTemplate.execute(status -> {
|
||||||
run.setStatus(TrainingStatus.DONE);
|
run.setStatus(TrainingStatus.DONE);
|
||||||
run.setCompletedAt(Instant.now());
|
run.setCompletedAt(Instant.now());
|
||||||
|
run.setCer(result.cer());
|
||||||
run.setLoss(result.loss());
|
run.setLoss(result.loss());
|
||||||
run.setAccuracy(result.accuracy());
|
run.setAccuracy(result.accuracy());
|
||||||
run.setEpochs(result.epochs());
|
run.setEpochs(result.epochs());
|
||||||
OcrTrainingRun updated = trainingRunRepository.save(run);
|
OcrTrainingRun updated = trainingRunRepository.save(run);
|
||||||
log.info("[trainingRun={}] Segmentation training completed — epochs={}", runId, result.epochs());
|
log.info("[trainingRun={}] Segmentation training completed — cer={} epochs={}", runId, result.cer(), result.epochs());
|
||||||
return updated;
|
return updated;
|
||||||
}));
|
}));
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
|
|||||||
Reference in New Issue
Block a user