From c2738835f0a2435ab03f411af3d168aec389d2a6 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 08 Dec 2015 01:18:04 +0000
Subject: [PATCH] Faster batch normalization
---
src/classifier.c | 5 +++++
1 files changed, 5 insertions(+), 0 deletions(-)
diff --git a/src/classifier.c b/src/classifier.c
index e243965..c0006e6 100644
--- a/src/classifier.c
+++ b/src/classifier.c
@@ -98,6 +98,11 @@
sprintf(buff, "%s/%s_%d.weights",backup_directory,base, epoch);
save_weights(net, buff);
}
+ if(*net.seen%1000 == 0){
+ char buff[256];
+ sprintf(buff, "%s/%s.backup",backup_directory,base);
+ save_weights(net, buff);
+ }
}
char buff[256];
sprintf(buff, "%s/%s.weights", backup_directory, base);
--
Gitblit v1.10.0