added conv1d, update to standardize_model

musco-ai / musco-pytorch

MUSCO: MUlti-Stage COmpression of neural networks

BSD 3-Clause "New" or "Revised" License

71 stars 16 forks source link

added conv1d, update to standardize_model #12

Open ABaaaC opened 3 years ago

ABaaaC commented 3 years ago

In standardize_model was added new feature: some attributes wasn't copy from CustomLayer to nn.Sequntial. Now it fixed

CP3DecomposedLayerConv1D was created in the same way as previous similar layers, but also it was added support of dilation of original convolution layer.