From d420edf57a2311fc002cbaacec48e3841fafa326 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 9 Jul 2019 17:08:11 -0400
Subject: [PATCH] fix
---
.../models_class1_pan_unselected/generate_hyperparameters.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py b/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py
index baeaa5bc..ccffc6b0 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,
--
GitLab