From 01affd07526622e07f51572785451fdf02019411 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sun, 28 Jan 2018 16:03:41 -0500
Subject: [PATCH] attempt fix deadlock
---
mhcflurry/train_allele_specific_models_command.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index b09194aa..8646a020 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -381,6 +381,9 @@ def calibrate_percentile_ranks(allele, predictor, peptides=None):
Private helper function.
"""
global GLOBAL_DATA
+ Class1NeuralNetwork.clear_model_cache()
+ import keras.backend as K
+ K.clear_session()
if peptides is None:
peptides = GLOBAL_DATA["calibration_peptides"]
if isinstance(predictor, str):
--
GitLab