如何在pytorch中使用可分離卷積 depth-wise Separable convolution

針對深度級別/可分離的卷積,可使用卷積組參數,若是groups = nInputPlane,就是Depthwise;若是groups = nInputPlane,kernel=(K, 1)(針對二維卷積,前面加上,groups=1 and kernel=(1, K)),就是可分離的。 如下將torch的官方手冊搬過來介紹使用方法(https://pytorch.org/docs/master/nn
相關文章
相關標籤/搜索