Update svd_layer.py

musco-ai · Feb 16, 2021 · 74b9f4a · 74b9f4a
1 parent e3e0822
commit 74b9f4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/musco/pytorch/compressor/decompositions/svd_layer.py b/musco/pytorch/compressor/decompositions/svd_layer.py
@@ -163,11 +163,11 @@ def __init__(self, layer, layer_name,
             self.rank = rank
         elif rank_selection == 'param_reduction':
             if  isinstance(self.layer, nn.Sequential):
-                prev_rank = self.layer[0].out_features
+                prev_rank = self.layer[0].out_channels
             else:
                 prev_rank = None
 
-            self.rank = estimate_rank_for_compression_rate((self.out_features, self.in_features),
+            self.rank = estimate_rank_for_compression_rate((self.out_channels, self.in_channels),
                                                            rate = param_reduction_rate,
                                                            key = 'svd',
                                                            prev_rank = prev_rank,