From c52fa4742866125e35e88d0653dac99473e5fffb Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 16 Apr 2018 10:09:10 +0000
Subject: [PATCH] Loss-graph store automatically (iterations == max_batches) at the end of training
---
src/image.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/image.c b/src/image.c
index 4821eff..61f760a 100644
--- a/src/image.c
+++ b/src/image.c
@@ -572,7 +572,7 @@
cvPutText(img, char_buff, pt1, &font, CV_RGB(0, 0, 0));
cvShowImage("average loss", img);
int k = cvWaitKey(20);
- if (k == 's') cvSaveImage("chart.jpg", img, 0);
+ if (k == 's' || current_batch == (max_batches-1)) cvSaveImage("chart.jpg", img, 0);
}
#endif // OPENCV
--
Gitblit v1.10.0