stuff

Joseph Redmon

2015-05-08 0cbfa4646128206300b9a30586615c3698abfb76

 src/network_kernels.cu

@@ -18,6 +18,7 @@
#include "normalization_layer.h"
#include "softmax_layer.h"
#include "dropout_layer.h"
#include "route_layer.h"
}

float * get_network_output_gpu_layer(network net, int i);
@@ -55,6 +56,9 @@
        else if(net.types[i] == CROP){
            forward_crop_layer_gpu(*(crop_layer *)net.layers[i], state);
        }
        else if(net.types[i] == ROUTE){
            forward_route_layer_gpu(*(route_layer *)net.layers[i], net);
        }
        state.input = get_network_output_gpu_layer(net, i);
    }
}
@@ -96,6 +100,9 @@
        else if(net.types[i] == SOFTMAX){
            backward_softmax_layer_gpu(*(softmax_layer *)net.layers[i], state);
        }
        else if(net.types[i] == ROUTE){
            backward_route_layer_gpu(*(route_layer *)net.layers[i], net);
        }
    }
}

@@ -142,6 +149,9 @@
    else if(net.types[i] == SOFTMAX){
        return ((softmax_layer *)net.layers[i]) -> output_gpu;
    }
    else if(net.types[i] == ROUTE){
        return ((route_layer *)net.layers[i]) -> output_gpu;
    }
    else if(net.types[i] == DROPOUT){
        return get_network_output_gpu_layer(net, i-1);
    }
@@ -170,6 +180,10 @@
        maxpool_layer layer = *(maxpool_layer *)net.layers[i];
        return layer.delta_gpu;
    }
    else if(net.types[i] == ROUTE){
        route_layer layer = *(route_layer *)net.layers[i];
        return layer.delta_gpu;
    }
    else if(net.types[i] == SOFTMAX){
        softmax_layer layer = *(softmax_layer *)net.layers[i];
        return layer.delta_gpu;