From 25f133d6efcdc1d2aa79263e4fe4880ca782c79e Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Tue, 11 Dec 2018 21:26:36 +0300 Subject: [PATCH] Another one minor fix --- src/convolutional_kernels.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/convolutional_kernels.cu b/src/convolutional_kernels.cu index af685631..7d4ef8f5 100644 --- a/src/convolutional_kernels.cu +++ b/src/convolutional_kernels.cu @@ -553,7 +553,7 @@ void backward_convolutional_layer_gpu(convolutional_layer l, network_state state state.workspace, l.workspace_size, &one, - l.dweightDesc, + l.dweightDesc16, l.weight_updates_gpu16); // l.weight_updates_gpu); cuda_convert_f16_to_f32(l.weight_updates_gpu16, l.c*l.n*l.size*l.size, l.weight_updates_gpu);