nn_decomposer does not support kernel_h and kernel_w
hahne opened this issue · 3 comments
hahne commented
nn_decomposer.py expects cur_layer_param.convolution_param.kernel_size._values to be set. This means the convolutional kernel needs to be square and kernel_h and kernel_w may not be different. Is there any special reason for this?
wenwei202 commented
@hahne There is no constraint on the kernel size from the algorithm perspective. It works in general (see here).
The code expects those, only because I did a lazy job to escape from a cumbersome code block of if-else-if-else to check that the kernel_size exists in kernel_size
or kernel_h kernel_w
. If it exists in kernel_size
, we also had to check it was a list or a scalar to make it work in general.
hahne commented
@wenwei202 ok understand. Thank you for the info!