diff --git a/mhcflurry/train_presentation_models_command.py b/mhcflurry/train_presentation_models_command.py index 8fcc47fce8c2cbf09ef00ed3c1b7042b63a4e4c4..b50795c3b0dfb32cfe6c8b4b701081605f460bc7 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(