From 8fd18add6e060a433629fae3fa2a7ef75df4644e Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 04 Nov 2015 03:23:42 +0000
Subject: [PATCH] CVPR Experiments
---
src/region_layer.c | 5 +++++
1 files changed, 5 insertions(+), 0 deletions(-)
diff --git a/src/region_layer.c b/src/region_layer.c
index 3239f87..3fff22b 100644
--- a/src/region_layer.c
+++ b/src/region_layer.c
@@ -226,6 +226,11 @@
void forward_region_layer_gpu(const region_layer l, network_state state)
{
+ if(!state.train){
+ copy_ongpu(l.batch*l.inputs, state.input, 1, l.output_gpu, 1);
+ return;
+ }
+
float *in_cpu = calloc(l.batch*l.inputs, sizeof(float));
float *truth_cpu = 0;
if(state.truth){
--
Gitblit v1.10.0