feng2022 commited on
Commit
5177a6b
·
1 Parent(s): e7d7286

Update Time_TravelRephotography/torch_utils/ops/bias_act.py

Browse files
Time_TravelRephotography/torch_utils/ops/bias_act.py CHANGED
@@ -14,6 +14,7 @@ import numpy as np
14
  import torch
15
  import dnnlib
16
  import traceback
 
17
 
18
  from .. import custom_ops
19
  from .. import misc
 
14
  import torch
15
  import dnnlib
16
  import traceback
17
+ from torch import nn
18
 
19
  from .. import custom_ops
20
  from .. import misc