From 451ef0a0a6b595bb8e4a945633659b4d31f0a372 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Mon, 20 Apr 2015 15:46:12 +0000 Subject: [PATCH] It's time, to du-du-du-du-DU-DU-DUEL!! https://www.youtube.com/watch?v=IVmtUK_1jh4 --- src/convolutional_layer.h | 4 ++-- 1 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/convolutional_layer.h b/src/convolutional_layer.h index eaf1562..5cf8adc 100644 --- a/src/convolutional_layer.h +++ b/src/convolutional_layer.h @@ -41,7 +41,7 @@ #ifdef GPU void forward_convolutional_layer_gpu(convolutional_layer layer, network_state state); void backward_convolutional_layer_gpu(convolutional_layer layer, network_state state); -void update_convolutional_layer_gpu(convolutional_layer layer, float learning_rate, float momentum, float decay); +void update_convolutional_layer_gpu(convolutional_layer layer, int batch, float learning_rate, float momentum, float decay); void push_convolutional_layer(convolutional_layer layer); void pull_convolutional_layer(convolutional_layer layer); @@ -53,7 +53,7 @@ convolutional_layer *make_convolutional_layer(int batch, int h, int w, int c, int n, int size, int stride, int pad, ACTIVATION activation); void resize_convolutional_layer(convolutional_layer *layer, int h, int w); void forward_convolutional_layer(const convolutional_layer layer, network_state state); -void update_convolutional_layer(convolutional_layer layer, float learning_rate, float momentum, float decay); +void update_convolutional_layer(convolutional_layer layer, int batch, float learning_rate, float momentum, float decay); image *visualize_convolutional_layer(convolutional_layer layer, char *window, image *prev_filters); void backward_convolutional_layer(convolutional_layer layer, network_state state); -- Gitblit v1.10.0