From 2b2441313b73c460a60c013c3b7bf9e19c994b6b Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 30 Oct 2014 18:28:37 +0000
Subject: [PATCH] col2im maybe a little faster
---
src/cnn.c | 13 ++++++++++++-
1 files changed, 12 insertions(+), 1 deletions(-)
diff --git a/src/cnn.c b/src/cnn.c
index de37bc3..ed5fee3 100644
--- a/src/cnn.c
+++ b/src/cnn.c
@@ -308,7 +308,7 @@
void train_imagenet()
{
- network net = parse_network_cfg("cfg/imagenet_backup_slowest_2340.cfg");
+ network net = parse_network_cfg("cfg/imagenet_small_830.cfg");
printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
int imgs = 1000/net.batch+1;
srand(6472345);
@@ -1016,6 +1016,17 @@
int main(int argc, char *argv[])
{
+ int i;
+ int ksize = 3;
+ int stride = 4;
+ int width_col = 20;
+ for(i = 0; i < 10; ++i){
+ int start = (i<ksize)?0:(i-ksize)/stride + 1;
+ int start2 = (i-ksize+stride)/stride;
+ int end = i/stride + 1;
+ end = (width_col < end) ? width_col : end;
+ printf("%d: %d vs %d, %d\n", i, start,start2, end);
+ }
if(argc != 2){
fprintf(stderr, "usage: %s <function>\n", argv[0]);
return 0;
--
Gitblit v1.10.0