Update svd_layer.py

musco-ai · Feb 16, 2021 · e3e0822 · e3e0822
1 parent 8dfa2a1
commit e3e0822
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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):