From 73f7aacf35ec9b1d0f9de9ddf38af0889f213e99 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 20 Sep 2016 18:34:49 +0000
Subject: [PATCH] better multigpu
---
src/parser.c | 4 ++++
1 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 3551983..2b285b5 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -954,7 +954,9 @@
void save_weights_upto(network net, char *filename, int cutoff)
{
#ifdef GPU
+ if(net.gpu_index >= 0){
cuda_set_device(net.gpu_index);
+ }
#endif
fprintf(stderr, "Saving weights to %s\n", filename);
FILE *fp = fopen(filename, "w");
@@ -1120,7 +1122,9 @@
void load_weights_upto(network *net, char *filename, int cutoff)
{
#ifdef GPU
+ if(net->gpu_index >= 0){
cuda_set_device(net->gpu_index);
+ }
#endif
fprintf(stderr, "Loading weights from %s...", filename);
fflush(stdout);
--
Gitblit v1.10.0