From 033e934ce82826c73d851098baf7ce4b1a27c89a Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 21 Feb 2018 16:14:01 +0000
Subject: [PATCH] If there is excessive GPU-RAM consumption by CUDNN then then do not use Workspace
---
src/route_layer.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/route_layer.c b/src/route_layer.c
index d18427a..dce7118 100644
--- a/src/route_layer.c
+++ b/src/route_layer.c
@@ -5,7 +5,7 @@
route_layer make_route_layer(int batch, int n, int *input_layers, int *input_sizes)
{
- fprintf(stderr,"Route Layer:");
+ fprintf(stderr,"route ");
route_layer l = {0};
l.type = ROUTE;
l.batch = batch;
--
Gitblit v1.10.0