Update svf.py

syp2ysy · Sep 27, 2022 · 0458898 · 0458898
1 parent 4a51e8f
commit 0458898
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/svf.py b/svf.py
@@ -52,7 +52,7 @@ def forward(self, x):
 
 
 full2low_mapping_n = {
-    nn.Conv2d: SVD_Conv2d,
+    nn.Conv2D: SVD_Conv2d,
     nn.Linear: SVD_Linear
 }