CoAtNet

drmouse-dlc · Oct 20, 2021 · 6c229ea · 6c229ea
1 parent 920734e
commit 6c229ea
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/model/attention/CoAtNet.py b/model/attention/CoAtNet.py
@@ -3,8 +3,8 @@
 import sys
 from math import sqrt
 sys.path.append('.')
-from conv.MBConv import MBConvBlock
-from attention.SelfAttention import ScaledDotProductAttention
+from model.conv.MBConv import MBConvBlock
+from model.attention.SelfAttention import ScaledDotProductAttention
 
 class CoAtNet(nn.Module):
     def __init__(self,in_ch,image_size,out_chs=[64,96,192,384,768]):

diff --git a/model/attention/DANet.py b/model/attention/DANet.py
@@ -2,8 +2,8 @@
 import torch
 from torch import nn
 from torch.nn import init
-from .SelfAttention import ScaledDotProductAttention
-from .SimplifiedSelfAttention import SimplifiedScaledDotProductAttention
+from model.attention.SelfAttention import ScaledDotProductAttention
+from model.attention.SimplifiedSelfAttention import SimplifiedScaledDotProductAttention
 
 class PositionAttentionModule(nn.Module):
 

diff --git a/model/attention/__pycache__/SelfAttention.cpython-38.pyc b/model/attention/__pycache__/SelfAttention.cpython-38.pyc
diff --git a/model/conv/__pycache__/MBConv.cpython-38.pyc b/model/conv/__pycache__/MBConv.cpython-38.pyc