diff --git a/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py b/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py
index baeaa5bc06fa7242eb55770551aaf41045f5f7ed..ccffc6b082b783d6626dd92b3900270fbcc644c6 100644
--- a/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py
+++ b/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py
@@ -42,6 +42,7 @@ base_hyperparameters = {
     'train_data': {
         'pretrain': True,
         'pretrain_peptides_per_epoch': 1024,
+        'pretrain_steps_per_epoch': 16,
         'pretrain_patience': 10,
     },
     'validation_split': 0.1,