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/crop_layer.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/crop_layer.c b/src/crop_layer.c
index d9950d6..7b34084 100644
--- a/src/crop_layer.c
+++ b/src/crop_layer.c
@@ -33,7 +33,7 @@
l.output = calloc(crop_width*crop_height * c*batch, sizeof(float));
#ifdef GPU
l.output_gpu = cuda_make_array(l.output, crop_width*crop_height*c*batch);
- l.rand_gpu = cuda_make_array(0, l.batch*8);
+ l.rand_gpu = cuda_make_array(0, l.batch*8);
#endif
return l;
}
--
Gitblit v1.10.0