Minor fix - filename of chart (Loss & TopK) - training Classifier

5 years ago · 6033baabc9
parent 379952208e
commit 6033baabc9
1 changed files with 1 additions and 1 deletions
--- a/src/classifier.c
+++ b/src/classifier.c
@ -120,7 +120,7 @@ void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int *gpus,
    int number_of_lines = 100;
    int img_size = 1000;
    char windows_name[100];
-    sprintf(windows_name, "average loss (id:%d)", random_gen());
+    sprintf(windows_name, "chart_%s.png", base);
    if (!dontuse_opencv) img = draw_train_chart(windows_name, max_img_loss, net.max_batches, number_of_lines, img_size, dont_show, chart_path);
 #endif  //OPENCV