@@ -59,37 +59,41 @@ def runCNN(arg):
5959 createDir (model_config ['wdir' ]);
6060 #create working dir
6161
62- #learning rate, batch-size and momentum
63- lrate = LearningRate .get_instance (model_config ['l_rate_method' ],model_config ['l_rate' ]);
64- batch_size = model_config ['batch_size' ];
65- momentum = model_config ['momentum' ]
66-
6762 cnn = CNN (numpy_rng ,theano_rng ,conv_layer_configs = conv_layer_config , batch_size = batch_size ,
68- n_outs = model_config ['n_outs' ],hidden_layers_sizes = mlp_config ['layers' ], conv_activation = conv_activation ,
69- hidden_activation = hidden_activation ,use_fast = conv_config ['use_fast' ])
63+ n_outs = model_config ['n_outs' ],hidden_layers_sizes = mlp_config ['layers' ],
64+ conv_activation = conv_activation ,hidden_activation = hidden_activation ,
65+ use_fast = conv_config ['use_fast' ])
7066
71- train_sets , train_xy , train_x , train_y = read_dataset (data_spec ['training' ],model_config ['batch_size' ])
72- valid_sets , valid_xy , valid_x , valid_y = read_dataset (data_spec ['validation' ],model_config ['batch_size' ])
67+ if model_config ['processes' ]['finetuning' ]:
68+
69+ #learning rate, batch-size and momentum
70+ lrate = LearningRate .get_instance (model_config ['l_rate_method' ],model_config ['l_rate' ]);
71+ batch_size = model_config ['batch_size' ];
72+ momentum = model_config ['momentum' ]
73+
74+ train_sets , train_xy , train_x , train_y = read_dataset (data_spec ['training' ],model_config ['batch_size' ])
75+ valid_sets , valid_xy , valid_x , valid_y = read_dataset (data_spec ['validation' ],model_config ['batch_size' ])
76+
77+ err = fineTunning (cnn ,train_sets ,train_xy ,train_x ,train_y ,
78+ valid_sets ,valid_xy ,valid_x ,valid_y ,lrate ,momentum ,batch_size );
7379
74- err = fineTunning (cnn ,train_sets ,train_xy ,train_x ,train_y ,
75- valid_sets ,valid_xy ,valid_x ,valid_y ,lrate ,momentum ,batch_size );
76-
77- _cnn2file (cnn .layers [0 :cnn .conv_layer_num ], filename = model_config ['conv_output_file' ],activation = conv_config ['activation' ]);
78- _nnet2file (cnn .layers [cnn .conv_layer_num :], filename = model_config ['hidden_output_file' ],activation = mlp_config ['activation' ]);
7980
8081 ####################
8182 ## TESTING ##
8283 ####################
83- try :
84- test_sets , test_xy , test_x , test_y = read_dataset (data_spec ['testing' ],model_config ['batch_size' ])
85- except KeyError :
86- #raise e
87- logger .info ("No testing set:Skiping Testing" );
88- logger .info ("Finshed" )
89- sys .exit (0 )
84+ if model_config ['processes' ]['testing' ]:
85+ try :
86+ test_sets , test_xy , test_x , test_y = read_dataset (data_spec ['testing' ],model_config ['batch_size' ])
87+ except KeyError :
88+ #raise e
89+ logger .info ("No testing set:Skiping Testing" );
90+ logger .info ("Finshed" )
91+ sys .exit (0 )
9092
91- pred ,err = testing (cnn ,test_sets , test_xy , test_x , test_y ,batch_size )
93+ pred ,err = testing (cnn ,test_sets , test_xy , test_x , test_y ,batch_size )
9294
95+ _cnn2file (cnn .layers [0 :cnn .conv_layer_num ], filename = model_config ['conv_output_file' ],activation = conv_config ['activation' ]);
96+ _nnet2file (cnn .layers [cnn .conv_layer_num :], filename = model_config ['hidden_output_file' ],activation = mlp_config ['activation' ]);
9397
9498if __name__ == '__main__' :
9599 setLogger (level = "DEBUG" );
0 commit comments