From 0419c54042b7dcdde5ba7857b6d0f91b8fcc2003 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sun, 07 Jan 2018 00:30:51 +0000
Subject: [PATCH] Tracking and Streaming fixed.
---
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 313cd6d..7a261c5 100644
--- a/src/network_kernels.cu
+++ b/src/network_kernels.cu
@@ -64,6 +64,7 @@
for(i = net.n-1; i >= 0; --i){
state.index = i;
layer l = net.layers[i];
+ if (l.stopbackward) break;
if(i == 0){
state.input = original_input;
state.delta = original_delta;
--
Gitblit v1.10.0