NIGHTMARE!!!!

Joseph Redmon

2015-07-08 a08ef29e085236f13fd04246c6774f9682a56208

 src/network_kernels.cu

@@ -7,6 +7,7 @@
#include "data.h"
#include "utils.h"
#include "params.h"
#include "parser.h"

#include "crop_layer.h"
#include "connected_layer.h"
@@ -58,11 +59,12 @@
{
    int i;
    float * original_input = state.input;
    float * original_delta = state.delta;
    for(i = net.n-1; i >= 0; --i){
        layer l = net.layers[i];
        if(i == 0){
            state.input = original_input;
            state.delta = 0;
            state.delta = original_delta;
        }else{
            layer prev = net.layers[i-1];
            state.input = prev.output_gpu;
@@ -119,6 +121,7 @@
        cuda_push_array(*net.truth_gpu, y, y_size);
    }
    state.input = *net.input_gpu;
    state.delta = 0;
    state.truth = *net.truth_gpu;
    state.train = 1;
    forward_network_gpu(net, state);
@@ -132,20 +135,18 @@
float *get_network_output_layer_gpu(network net, int i)
{
    layer l = net.layers[i];
    cuda_pull_array(l.output_gpu, l.output, l.outputs*l.batch);
    if(l.type == CONVOLUTIONAL){
        return l.output;
    } else if(l.type == DECONVOLUTIONAL){
        return l.output;
    } else if(l.type == CONNECTED){
        cuda_pull_array(l.output_gpu, l.output, l.outputs*l.batch);
        return l.output;
    } else if(l.type == DETECTION){
        cuda_pull_array(l.output_gpu, l.output, l.outputs*l.batch);
        return l.output;
    } else if(l.type == MAXPOOL){
        return l.output;
    } else if(l.type == SOFTMAX){
        pull_softmax_layer_output(l);
        return l.output;
    }
    return 0;