Hello: The kernel size of the last layer in the code is 1, which may become larger in practice depending on the cls_module result?