Minor changes to seq2seq attention

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(