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/dropout_layer_kernels.cu | 1 - 1 files changed, 0 insertions(+), 1 deletions(-) diff --git a/src/dropout_layer_kernels.cu b/src/dropout_layer_kernels.cu index f65e3c3..7e51bd5 100644 --- a/src/dropout_layer_kernels.cu +++ b/src/dropout_layer_kernels.cu @@ -6,7 +6,6 @@ #include "dropout_layer.h" #include "cuda.h" #include "utils.h" -#include "params.h" } __global__ void yoloswag420blazeit360noscope(float *input, int size, float *rand, float prob, float scale) -- Gitblit v1.10.0