From 4ea2b240d8cf3d27690ff5cbe541c68ed5463505 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 9 Feb 2018 18:31:53 -0500
Subject: [PATCH] mixed gpu / cpu training hack
---
mhcflurry/train_allele_specific_models_command.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index 9396230b..07025dac 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -194,9 +194,10 @@ def run(argv=sys.argv[1:]):
# environment variable. To do this, we push environment variables
# onto a queue. Each worker reads a single item from the queue,
# which is a list of environment variables to set.
+ cpus = 16
next_device = itertools.cycle([
"%d" % num for num in range(args.gpus)
- ])
+ ] + ["" for num in range(cpus)])
env_queue = Queue()
for num in range(args.num_jobs[0]):
item = [
--
GitLab