From 62235e9aa3d0c15d87d49bf340625d075cba3e65 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Sat, 19 Nov 2016 05:51:36 +0000 Subject: [PATCH] cpu batch norm works --- src/utils.h | 8 +++++++- 1 files changed, 7 insertions(+), 1 deletions(-) diff --git a/src/utils.h b/src/utils.h index 7e49818..bbc6765 100644 --- a/src/utils.h +++ b/src/utils.h @@ -5,7 +5,9 @@ #include "list.h" #define SECRET_NUM -1234 +#define TWO_PI 6.2831853071795864769252866 +int *read_map(char *filename); void shuffle(void *arr, size_t n, size_t size); void sorta_shuffle(void *arr, size_t n, size_t size, size_t sections); void free_ptrs(void **ptrs, int n); @@ -18,7 +20,7 @@ void write_all(int fd, char *buffer, size_t bytes); int read_all_fail(int fd, char *buffer, size_t bytes); int write_all_fail(int fd, char *buffer, size_t bytes); -char *find_replace(char *str, char *orig, char *rep); +void find_replace(char *str, char *orig, char *rep, char *output); void error(const char *s); void malloc_error(); void file_error(char *s); @@ -36,16 +38,19 @@ void translate_array(float *a, int n, float s); int max_index(float *a, int n); float constrain(float min, float max, float a); +int constrain_int(int a, int min, int max); float mse_array(float *a, int n); float rand_normal(); size_t rand_size_t(); float rand_uniform(float min, float max); +float rand_scale(float s); int rand_int(int min, int max); float sum_array(float *a, int n); float mean_array(float *a, int n); void mean_arrays(float **a, int n, int els, float *avg); float variance_array(float *a, int n); float mag_array(float *a, int n); +float dist_array(float *a, float *b, int n, int sub); float **one_hot_encode(float *a, int n, int k); float sec(clock_t clocks); int find_int_arg(int argc, char **argv, char *arg, int def); @@ -53,6 +58,7 @@ int find_arg(int argc, char* argv[], char *arg); char *find_char_arg(int argc, char **argv, char *arg, char *def); int sample_array(float *a, int n); +void print_statistics(float *a, int n); #endif -- Gitblit v1.10.0