rinong commited on
Commit
a390b6f
1 Parent(s): 558a786

Fixed missing op import

Browse files
Files changed (1) hide show
  1. model/sg2_model.py +2 -0
model/sg2_model.py CHANGED
@@ -8,6 +8,8 @@ from torch import nn
8
  from torch.nn import functional as F
9
  from torch.autograd import Function
10
 
 
 
11
  if torch.cuda.is_available():
12
  from op.fused_act import FusedLeakyReLU, fused_leaky_relu
13
  from op.upfirdn2d import upfirdn2d
 
8
  from torch.nn import functional as F
9
  from torch.autograd import Function
10
 
11
+ from op import conv2d_gradfix
12
+
13
  if torch.cuda.is_available():
14
  from op.fused_act import FusedLeakyReLU, fused_leaky_relu
15
  from op.upfirdn2d import upfirdn2d