From bc810016a1f2eadf33a1ac800b64962a42f3d402 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 02 Aug 2017 22:36:22 +0000
Subject: [PATCH] cuDNN 6.0 supported. Also speed of console example improved.
---
src/softmax_layer.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/softmax_layer.c b/src/softmax_layer.c
index 0aa9047..5d15314 100644
--- a/src/softmax_layer.c
+++ b/src/softmax_layer.c
@@ -10,7 +10,7 @@
softmax_layer make_softmax_layer(int batch, int inputs, int groups)
{
assert(inputs%groups == 0);
- fprintf(stderr, "Softmax Layer: %d inputs\n", inputs);
+ fprintf(stderr, "softmax %4d\n", inputs);
softmax_layer l = {0};
l.type = SOFTMAX;
l.batch = batch;
--
Gitblit v1.10.0