diff --git a/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py b/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py
index 4ed6e35cacec09ce3a460bb19d1cb890c39ff5f3..676d2006a455e83cc23d7d89ccc07b8867a2659a 100644
--- a/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py
+++ b/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py
@@ -42,9 +42,9 @@ base_hyperparameters = {
     'random_negative_rate': 0.2,
     'train_data': {
         'pretrain': True,
-        'pretrain_peptides_per_epoch': 1024,
-        'pretrain_steps_per_epoch': 16,
-        'pretrain_patience': 10,
+        'pretrain_peptides_per_epoch': 64,
+        'pretrain_steps_per_epoch': 256,
+        'pretrain_patience': 5,
         'pretrain_min_delta': 0.0001,
         'pretrain_max_val_loss': 0.10,
     },