From dedb0adfb0a9a722d1603b86421914fc8e2d7f56 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Mon, 15 Jul 2019 17:35:51 -0400
Subject: [PATCH] fix
---
.../generate_hyperparameters.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py b/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py
index 4ed6e35c..676d2006 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,
},
--
GitLab