From 283333b52fc07a0e6db489cf56bc8f791bea9b00 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Thu, 9 Jan 2020 19:44:08 +0300 Subject: [PATCH] Added network resizing for SAM layer --- src/network.c | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/network.c b/src/network.c index 0de38420..fd5eea84 100644 --- a/src/network.c +++ b/src/network.c @@ -541,6 +541,8 @@ int resize_network(network *net, int w, int h) resize_shortcut_layer(&l, w, h, net); }else if (l.type == SCALE_CHANNELS) { resize_scale_channels_layer(&l, net); + }else if (l.type == SAM) { + resize_sam_layer(&l, w, h); }else if (l.type == DROPOUT) { resize_dropout_layer(&l, inputs); l.out_w = l.w = w;