From 0368d65eb8ca00bda5c7289cb382a2197ede628c Mon Sep 17 00:00:00 2001
From: Volkan Vural <volkanvural@gmail.com>
Date: Mon, 4 Nov 2019 12:47:09 -0800
Subject: [PATCH] Update experiments/mem_allJobs_m50_h20_r50.py
---
experiments/mem_allJobs_m50_h20_r50.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/experiments/mem_allJobs_m50_h20_r50.py b/experiments/mem_allJobs_m50_h20_r50.py
index aa1cd07..23a7180 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
--
GitLab