| | |
|---|
| | | SOFTMAX, |
|---|
| | | NORMALIZATION, |
|---|
| | | DROPOUT, |
|---|
| | | CROP |
|---|
| | | FREEWEIGHT, |
|---|
| | | CROP, |
|---|
| | | COST |
|---|
| | | } LAYER_TYPE; |
|---|
| | | |
|---|
| | | typedef struct { |
|---|
| | |
|---|
| | | float *output; |
|---|
| | | |
|---|
| | | #ifdef GPU |
|---|
| | | cl_mem input_cl; |
|---|
| | | cl_mem output_cl; |
|---|
| | | cl_mem *input_cl; |
|---|
| | | cl_mem *truth_cl; |
|---|
| | | #endif |
|---|
| | | } network; |
|---|
| | | |
|---|
| | | #ifdef GPU |
|---|
| | | void forward_network_gpu(network net, cl_mem input, int train); |
|---|
| | | #ifndef GPU |
|---|
| | | typedef int cl_mem; |
|---|
| | | #endif |
|---|
| | | |
|---|
| | | cl_mem get_network_output_cl_layer(network net, int i); |
|---|
| | | cl_mem get_network_delta_cl_layer(network net, int i); |
|---|
| | | void forward_network_gpu(network net, cl_mem input, cl_mem truth, int train); |
|---|
| | | void backward_network_gpu(network net, cl_mem input); |
|---|
| | | void update_network_gpu(network net); |
|---|
| | | float train_network_sgd_gpu(network net, data d, int n); |
|---|
| | | float train_network_data_gpu(network net, data d, int n); |
|---|
| | | float *network_predict_gpu(network net, float *input); |
|---|
| | | float network_accuracy_gpu(network net, data d); |
|---|
| | | float *network_accuracies_gpu(network net, data d); |
|---|
| | | |
|---|
| | | float *network_accuracies(network net, data d); |
|---|
| | | |
|---|
| | | network make_network(int n, int batch); |
|---|
| | | void forward_network(network net, float *input, int train); |
|---|
| | | float backward_network(network net, float *input, float *truth); |
|---|
| | | void forward_network(network net, float *input, float *truth, int train); |
|---|
| | | void backward_network(network net, float *input); |
|---|
| | | void update_network(network net); |
|---|
| | | float train_network_sgd(network net, data d, int n); |
|---|
| | | float train_network_batch(network net, data d, int n); |
|---|
| | | float train_network_data_cpu(network net, data d, int n); |
|---|
| | | void train_network(network net, data d); |
|---|
| | | matrix network_predict_data(network net, data test); |
|---|
| | | float *network_predict(network net, float *input); |
|---|
| | | float network_accuracy(network net, data d); |
|---|
| | | float network_accuracy_multi(network net, data d, int n); |
|---|
| | | void top_predictions(network net, int n, int *index); |
|---|
| | | float *get_network_output(network net); |
|---|
| | | float *get_network_output_layer(network net, int i); |
|---|
| | | float *get_network_delta_layer(network net, int i); |
|---|
| | |
|---|
| | | void print_network(network net); |
|---|
| | | void visualize_network(network net); |
|---|
| | | int resize_network(network net, int h, int w, int c); |
|---|
| | | void set_batch_network(network *net, int b); |
|---|
| | | void set_learning_network(network *net, float rate, float momentum, float decay); |
|---|
| | | int get_network_input_size(network net); |
|---|
| | | float get_network_cost(network net); |
|---|
| | | |
|---|
| | | #endif |
|---|
| | | |
|---|
