Alpha Release

Author: sudhargk

io_modules/model_io.py

@@ -94,7 +94,6 @@ def _cnn2file(conv_layers,mlp_layers,filename='nnet.out',
 	   if i == 0:
 		   mlp_dict[dict_a] = array_2_string(input_factor * mlp_layers[i].params[0].get_value())
 	   else:
-		   print mlp_layers[i].params[0]
 		   mlp_dict[dict_a] = array_2_string(factor * mlp_layers[i].params[0].get_value())
 	   dict_a = str(i) + ' b'
 	   mlp_dict[dict_a] = array_2_string(mlp_layers[i].params[1].get_value())

run/run_CNN.py

@@ -92,6 +92,9 @@ def runCNN(arg):
 
 		pred,err=testing(cnn,test_sets, test_xy, test_x, test_y,batch_size)
 
+
+	_cnn2file(cnn.layers[0:cnn.conv_layer_num],cnn.layers[cnn.conv_layer_num:], filename=model_config['output_file']);
+	
 	##########################
 	##	Export Features ##
 	##########################