feng2022 commited on
Commit
0f60bc8
1 Parent(s): a5ea32b

Update Time_TravelRephotography/model.py

Browse files
Files changed (1) hide show
  1. Time_TravelRephotography/model.py +1 -1
Time_TravelRephotography/model.py CHANGED
@@ -608,7 +608,7 @@ class ConvLayer(nn.Sequential):
608
 
609
  if activate:
610
  if bias:
611
- layers.append(bias_act(out_channel))
612
 
613
  else:
614
  layers.append(ScaledLeakyReLU(0.2))
 
608
 
609
  if activate:
610
  if bias:
611
+ layers.append(bias_act_relu(out_channel))
612
 
613
  else:
614
  layers.append(ScaledLeakyReLU(0.2))