From cdd1cb0e8c4fda3671714bb5ad6ba1825cff16d1 Mon Sep 17 00:00:00 2001
From: Puneet Kohli <punkohl@gmail.com>
Date: Sun, 06 May 2018 21:45:26 +0000
Subject: [PATCH] Change matplotlib backend to 'agg' when 'show_plot' is false. This allows the script to be run from command line without any errors (for ex, on a remote machine)

---
 src/avgpool_layer.c |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/src/avgpool_layer.c b/src/avgpool_layer.c
index c6db477..b6932fe 100644
--- a/src/avgpool_layer.c
+++ b/src/avgpool_layer.c
@@ -4,7 +4,7 @@
 
 avgpool_layer make_avgpool_layer(int batch, int w, int h, int c)
 {
-    fprintf(stderr, "Avgpool Layer: %d x %d x %d image\n", w,h,c);
+    fprintf(stderr, "avg                     %4d x%4d x%4d   ->  %4d\n",  w, h, c, c);
     avgpool_layer l = {0};
     l.type = AVGPOOL;
     l.batch = batch;

--
Gitblit v1.10.0