feng2022 commited on
Commit
4326ba1
1 Parent(s): 1dce2b5

Update Time_TravelRephotography/model.py

Browse files
Files changed (1) hide show
  1. Time_TravelRephotography/model.py +1 -1
Time_TravelRephotography/model.py CHANGED
@@ -333,7 +333,7 @@ class StyledConv(nn.Module):
333
  self.noise = NoiseInjection()
334
  # self.bias = nn.Parameter(torch.zeros(1, out_channel, 1, 1))
335
  # self.activate = ScaledLeakyReLU(0.2)
336
- self.activate =torch_utils.ops.bias_act._bias_act_ref(out_channel)
337
 
338
  def forward(self, input, style, noise=None):
339
  out = self.conv(input, style)
 
333
  self.noise = NoiseInjection()
334
  # self.bias = nn.Parameter(torch.zeros(1, out_channel, 1, 1))
335
  # self.activate = ScaledLeakyReLU(0.2)
336
+ self.activate =_bias_act_ref(out_channel)
337
 
338
  def forward(self, input, style, noise=None):
339
  out = self.conv(input, style)