diff --git a/musco/pytorch/compressor/decompositions/svd_layer.py b/musco/pytorch/compressor/decompositions/svd_layer.py index 5f7ce15..0b1648a 100644 --- a/musco/pytorch/compressor/decompositions/svd_layer.py +++ b/musco/pytorch/compressor/decompositions/svd_layer.py @@ -136,6 +136,8 @@ def __init__(self, layer, layer_name, self.layer = layer self.pretrained = pretrained + self.min_rank = 2 + #print(layer) if isinstance(self.layer, nn.Sequential):