Update conv2d_gradfix.py

williamyang1991 · web-flow · commit fcc547c7728b · 2022-07-18T16:35:38.000+08:00
diff --git a/model/stylegan/op_cpu/conv2d_gradfix.py b/model/stylegan/op_cpu/conv2d_gradfix.py
@@ -1,3 +1,4 @@
+import contextlib
 import warnings
 
 import torch
