stuff

Joseph Redmon

2015-05-08 0cbfa4646128206300b9a30586615c3698abfb76

 src/cost_layer.h

@@ -1,8 +1,9 @@
#ifndef COST_LAYER_H
#define COST_LAYER_H
#include "params.h"

typedef enum{
    SSE
    SSE, MASKED
} COST_TYPE;

typedef struct {
@@ -21,12 +22,12 @@
COST_TYPE get_cost_type(char *s);
char *get_cost_string(COST_TYPE a);
cost_layer *make_cost_layer(int batch, int inputs, COST_TYPE type);
void forward_cost_layer(const cost_layer layer, float *input, float *truth);
void backward_cost_layer(const cost_layer layer, float *input, float *delta);
void forward_cost_layer(const cost_layer layer, network_state state);
void backward_cost_layer(const cost_layer layer, network_state state);

#ifdef GPU
void forward_cost_layer_gpu(cost_layer layer, float * input, float * truth);
void backward_cost_layer_gpu(const cost_layer layer, float * input, float * delta);
void forward_cost_layer_gpu(cost_layer layer, network_state state);
void backward_cost_layer_gpu(const cost_layer layer, network_state state);
#endif

#endif