diff --git a/experiments/mem_allJobs_m50_h20_r50.py b/experiments/mem_allJobs_m50_h20_r50.py index aa1cd07c87e5ff1981010862683a285bec100ec2..23a7180d39e1108646f6ab6699bb0e7b6506caaa 100644 --- a/experiments/mem_allJobs_m50_h20_r50.py +++ b/experiments/mem_allJobs_m50_h20_r50.py @@ -55,7 +55,7 @@ from keras import backend as K K.tensorflow_backend._get_available_gpus() -config = tf.ConfigProto( device_count = {'GPU': 4 , 'CPU': 2} ) +config = tf.ConfigProto( device_count = {'GPU': 1 , 'CPU': 2} ) sess = tf.Session(config=config) keras.backend.set_session(sess) @@ -483,9 +483,9 @@ def dict_product(dicts): hyperparameters = { 'hidden_size':[2,4,8,16,32,64,128], 'lr': [0.1,0.01,0.001,0.0001], - 'dropout': [0,0.1,0.2,0.3,0.4], + 'dropout': [0,0.05,0.1,0.2,0.3,0.4], 'epochs' : [25, 50, 100, 200, 400], - 'batch_size' : [5,10,25, 50, 100, 200] + 'batch_size' : [5,25, 50, 100, 200,1000,5000] } # Build list of hyperparameter combinations