From ce757768cf0165ce81dd97c4a1bf4be9d7bbd397 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 11 Oct 2019 11:10:07 -0400
Subject: [PATCH] fix
---
mhcflurry/train_pan_allele_models_command.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/mhcflurry/train_pan_allele_models_command.py b/mhcflurry/train_pan_allele_models_command.py
index 7f3b67db..2334850b 100644
--- a/mhcflurry/train_pan_allele_models_command.py
+++ b/mhcflurry/train_pan_allele_models_command.py
@@ -423,7 +423,8 @@ def train_models(args):
global GLOBAL_DATA
print("Beginning training.")
- predictor = Class1AffinityPredictor.load(args.out_models_dir)
+ predictor = Class1AffinityPredictor.load(
+ args.out_models_dir, optimization_level=0)
print("Loaded predictor with %d networks" % len(predictor.neural_networks))
with open(join(args.out_models_dir, "training_init_info.pkl"), "rb") as fd:
--
GitLab