:charizard: :charizard: :charizard:

Joseph Redmon

2016-11-06 252e3b1916cfaca0783c9e90efaa55eb07b1a8cd

 src/dropout_layer.c

@@ -1,5 +1,4 @@
#include "dropout_layer.h"
#include "params.h"
#include "utils.h"
#include "cuda.h"
#include <stdlib.h>
@@ -16,7 +15,11 @@
    l.batch = batch;
    l.rand = calloc(inputs*batch, sizeof(float));
    l.scale = 1./(1.-probability);
    l.forward = forward_dropout_layer;
    l.backward = backward_dropout_layer;
    #ifdef GPU
    l.forward_gpu = forward_dropout_layer_gpu;
    l.backward_gpu = backward_dropout_layer_gpu;
    l.rand_gpu = cuda_make_array(l.rand, inputs*batch);
    #endif
    return l;