mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

accidentally broke multigpu a while ago :-/

Browse Source
This commit is contained in:
Joseph Redmon
2017-06-26 14:00:18 -07:00
parent 63ebd694d7
commit 59ed1719d4
12 changed files with 119 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/darknet.c
View File

@@ -337,7 +337,7 @@ void denormalize_net(char *cfgfile, char *weightfile, char *outfile)
int i;
for (i = 0; i < net.n; ++i) {
layer l = net.layers[i];
if (l.type == CONVOLUTIONAL && l.batch_normalize) {
if ((l.type == DECONVOLUTIONAL || l.type == CONVOLUTIONAL) && l.batch_normalize) {
denormalize_convolutional_layer(l);
net.layers[i].batch_normalize=0;
}