From 64e7f0449d294564231a14535d3edef695e8ee82 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Mon, 27 Jan 2020 17:41:55 -0500
Subject: [PATCH] fix
---
mhcflurry/train_presentation_models_command.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/mhcflurry/train_presentation_models_command.py b/mhcflurry/train_presentation_models_command.py
index 8fcc47fc..b50795c3 100644
--- a/mhcflurry/train_presentation_models_command.py
+++ b/mhcflurry/train_presentation_models_command.py
@@ -105,7 +105,9 @@ def main(args):
os.mkdir(args.out_models_dir)
print("Done.")
- affinity_predictor = Class1AffinityPredictor.load(args.affinity_predictor)
+ affinity_predictor = Class1AffinityPredictor.load(
+ args.affinity_predictor,
+ optimization_level=0)
processing_predictor_with_flanks = Class1ProcessingPredictor.load(
args.processing_predictor_with_flanks)
processing_predictor_without_flanks = Class1ProcessingPredictor.load(
--
GitLab