From 192135106f090182e08d9ad134cb0e4668c3eb6e Mon Sep 17 00:00:00 2001
From: stereomatchingkiss <thamngapwei@gmail.com>
Date: Sat, 03 Feb 2018 10:30:42 +0000
Subject: [PATCH] fix bug 368
---
src/network.c | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index 79940e6..56a316c 100644
--- a/src/network.c
+++ b/src/network.c
@@ -218,6 +218,7 @@
state.delta = prev.delta;
}
layer l = net.layers[i];
+ if (l.stopbackward) break;
l.backward(l, state);
}
}
@@ -356,6 +357,7 @@
}else if(l.type == COST){
resize_cost_layer(&l, inputs);
}else{
+ fprintf(stderr, "Resizing type %d \n", (int)l.type);
error("Cannot resize this type of layer");
}
if(l.workspace_size > workspace_size) workspace_size = l.workspace_size;
--
Gitblit v1.10.0