diff --git a/code_jurica/seq2seq_attention.py b/code_jurica/seq2seq_attention.py
index 4a87398c5afa35e3dba36b19762063b9cf021a3b..67c81e47e8d95607737536e39de72f58fa91d20b 100644
--- a/code_jurica/seq2seq_attention.py
+++ b/code_jurica/seq2seq_attention.py
@@ -34,6 +34,7 @@ tf.set_random_seed(1234)
 # Don't pre-allocate memory; allocate as-needed
 config.gpu_options.allow_growth = True
 config.gpu_options.allocator_type = 'BFC'
+config.gpu_options.visible_device_list = "2"
 sess = tf.Session(graph=tf.get_default_graph(), config=config)
 K.set_session(sess)
 
@@ -55,7 +56,7 @@ with open('models/icd10_mappings.p', 'rb') as handle:
 callbacks_list = [
     EarlyStopping(
         monitor='val_loss',
-        patience=5,
+        patience=50,
         # min_delta=0.001
     ),
     ModelCheckpoint(