danifei commited on
Commit
e1b9be9
·
verified ·
1 Parent(s): 4219c5e

Update archs/NAFBlock.py

Browse files
Files changed (1) hide show
  1. archs/NAFBlock.py +2 -2
archs/NAFBlock.py CHANGED
@@ -4,8 +4,8 @@ import torch.nn.functional as F
4
 
5
  # Modules from model
6
  try:
7
- from .arch_util import LayerNorm2d
8
- import .arch_util as arch_util
9
  except:
10
  from arch_util import LayerNorm2d
11
  import arch_util as arch_util
 
4
 
5
  # Modules from model
6
  try:
7
+ from archs.arch_util import LayerNorm2d
8
+ import archs.arch_util as arch_util
9
  except:
10
  from arch_util import LayerNorm2d
11
  import arch_util as arch_util