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 | 7 ++++++-
1 files changed, 6 insertions(+), 1 deletions(-)
diff --git a/src/classifier.c b/src/classifier.c
index 2f9a91f..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);
@@ -279,7 +284,7 @@
int i;
if (target_layer >= 0){
- layer l = net.layers[target_layer];
+ //layer l = net.layers[target_layer];
}
for(i = 0; i < val.X.rows; ++i){
--
Gitblit v1.10.0