Read and store networks in ascii instead of raw bytes, necessary for some machines.

Author: Ricardicus

CMakeLists.txt

@@ -1,5 +1,6 @@
 cmake_minimum_required (VERSION 2.6)
 add_compile_definitions(WINDOWS)
+add_compile_definitions(STORE_NET_AS_ASCII)
 add_compile_definitions(_CRT_SECURE_NO_WARNINGS)
 project (net)
 add_executable(net main.c layers.c lstm.c set.c utilities.c)

lstm.c

@@ -825,6 +825,19 @@ void lstm_store_net_layers(lstm_model_t** model, const char * filename, int laye
 
 	while ( p < layers ) {
 
+#ifdef STORE_NET_AS_ASCII
+		vector_store_ascii(model[p]->Wy, model[p]->F * model[p]->N, fp);
+		vector_store_ascii(model[p]->Wi, model[p]->N * model[p]->S, fp);
+		vector_store_ascii(model[p]->Wc, model[p]->N * model[p]->S, fp);
+		vector_store_ascii(model[p]->Wo, model[p]->N * model[p]->S, fp);
+		vector_store_ascii(model[p]->Wf, model[p]->N * model[p]->S, fp);
+
+		vector_store_ascii(model[p]->by, model[p]->F, fp);
+		vector_store_ascii(model[p]->bi, model[p]->N, fp);
+		vector_store_ascii(model[p]->bc, model[p]->N, fp);
+		vector_store_ascii(model[p]->bf, model[p]->N, fp);
+		vector_store_ascii(model[p]->bo, model[p]->N, fp);
+#else
 		vector_store(model[p]->Wy, model[p]->F * model[p]->N, fp);
 		vector_store(model[p]->Wi, model[p]->N * model[p]->S, fp);
 		vector_store(model[p]->Wc, model[p]->N * model[p]->S, fp);
@@ -836,6 +849,7 @@ void lstm_store_net_layers(lstm_model_t** model, const char * filename, int laye
 		vector_store(model[p]->bc, model[p]->N, fp);
 		vector_store(model[p]->bf, model[p]->N, fp);
 		vector_store(model[p]->bo, model[p]->N, fp);
+#endif
 
 		++p;
 	}
@@ -918,6 +932,19 @@ void lstm_read_net_layers(lstm_model_t** model, const char * filename, int layer
 
 	while ( p < layers ) {
 
+#ifdef STORE_NET_AS_ASCII
+		vector_read_ascii(model[p]->Wy, model[p]->F * model[p]->N, fp);
+		vector_read_ascii(model[p]->Wi, model[p]->N * model[p]->S, fp);
+		vector_read_ascii(model[p]->Wc, model[p]->N * model[p]->S, fp);
+		vector_read_ascii(model[p]->Wo, model[p]->N * model[p]->S, fp);
+		vector_read_ascii(model[p]->Wf, model[p]->N * model[p]->S, fp);
+
+		vector_read_ascii(model[p]->by, model[p]->F, fp);
+		vector_read_ascii(model[p]->bi, model[p]->N, fp);
+		vector_read_ascii(model[p]->bc, model[p]->N, fp);
+		vector_read_ascii(model[p]->bf, model[p]->N, fp);
+		vector_read_ascii(model[p]->bo, model[p]->N, fp);
+#else
 		vector_read(model[p]->Wy, model[p]->F * model[p]->N, fp);
 		vector_read(model[p]->Wi, model[p]->N * model[p]->S, fp);
 		vector_read(model[p]->Wc, model[p]->N * model[p]->S, fp);
@@ -929,6 +956,7 @@ void lstm_read_net_layers(lstm_model_t** model, const char * filename, int layer
 		vector_read(model[p]->bc, model[p]->N, fp);
 		vector_read(model[p]->bf, model[p]->N, fp);
 		vector_read(model[p]->bo, model[p]->N, fp);
+#endif
 
 		++p;	
 	}

utilities.c

@@ -546,6 +546,31 @@ void 	vector_read(double * V, int L, FILE * fp)
 
 }
 
+void 	vector_store_ascii(double* V, int L, FILE * fp)
+{
+	int l = 0;
+
+	while ( l < L ){
+		fprintf(fp, "%.20lf\r\n", V[l]);
+		++l;
+	}
+}
+
+void 	vector_read_ascii(double * V, int L, FILE * fp)
+{
+	int l = 0;
+
+	while ( l < L ) {
+		if ( fscanf(fp, "%lf", &V[l]) <= 0 ) {
+			fprintf(stderr, "%s.%s Failed to read file\r\n",
+				__FILE__, __func__);
+			exit(1);
+		}
+		++l;
+	}
+
+}
+
 /*
 * 	This function is used to store a JSON file representation
 * 	of a LSTM neural network that can be read by an HTML application.

utilities.h

@@ -82,5 +82,7 @@ void 	vector_print_min_max(char *, double *, int);
 void 	vector_read(double *, int, FILE *);
 void 	vector_store(double *, int, FILE *);
 void 	matrix_store(double **, int, int, FILE *);  
-void 	matrix_read(double **, int, int, FILE *);   
+void 	matrix_read(double **, int, int, FILE *);
+void 	vector_read_ascii(double *, int, FILE *);
+void 	vector_store_ascii(double *, int, FILE *);
 #endif