From 2f62fe33c913cd9484fe7f2486889d12292c66e0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 07 Feb 2015 02:53:53 +0000
Subject: [PATCH] saving weight files as binaries, hell yeah
---
src/network_kernels.cu | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/src/network_kernels.cu b/src/network_kernels.cu
index de8f659..c49f37b 100644
--- a/src/network_kernels.cu
+++ b/src/network_kernels.cu
@@ -61,6 +61,7 @@
forward_crop_layer_gpu(layer, train, input);
input = layer.output_gpu;
}
+ //cudaDeviceSynchronize();
//printf("Forward %d %s %f\n", i, get_layer_string(net.types[i]), sec(clock() - time));
}
}
--
Gitblit v1.10.0