From fc9b867dd9c9a6d38d7fe478217060e11b9e7e1b Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 16 Nov 2016 08:15:46 +0000
Subject: [PATCH] :fire: :fire: :dragonite:
---
src/detector.c | 6 +++++-
1 files changed, 5 insertions(+), 1 deletions(-)
diff --git a/src/detector.c b/src/detector.c
index 3853ebb..a513816 100644
--- a/src/detector.c
+++ b/src/detector.c
@@ -136,15 +136,19 @@
i = get_current_batch(net);
printf("%d: %f, %f avg, %f rate, %lf seconds, %d images\n", get_current_batch(net), loss, avg_loss, get_current_rate(net), sec(clock()-time), i*imgs);
- if(i%100==0 || (i < 1000 && i%100 == 0)){
+ if(i%1000==0 || (i < 1000 && i%100 == 0)){
+#ifdef GPU
if(ngpus != 1) sync_nets(nets, ngpus, 0);
+#endif
char buff[256];
sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
save_weights(net, buff);
}
free_data(train);
}
+#ifdef GPU
if(ngpus != 1) sync_nets(nets, ngpus, 0);
+#endif
char buff[256];
sprintf(buff, "%s/%s_final.weights", backup_directory, base);
save_weights(net, buff);
--
Gitblit v1.10.0