asdasdasdasd commited on
Commit
e72a8c5
·
1 Parent(s): cdbd3dd

Update model_core.py

Browse files
Files changed (1) hide show
  1. model_core.py +3 -3
model_core.py CHANGED
@@ -2,9 +2,9 @@ import torch
2
  import torch.nn as nn
3
  import torch.nn.functional as F
4
 
5
- from components.attention import ChannelAttention, SpatialAttention, DualCrossModalAttention
6
- from components.srm_conv import SRMConv2d_simple, SRMConv2d_Separate
7
- from networks.xception import TransferModel
8
 
9
 
10
  class SRMPixelAttention(nn.Module):
 
2
  import torch.nn as nn
3
  import torch.nn.functional as F
4
 
5
+ from attention import ChannelAttention, SpatialAttention, DualCrossModalAttention
6
+ from srm_conv import SRMConv2d_simple, SRMConv2d_Separate
7
+ from xception import TransferModel
8
 
9
 
10
  class SRMPixelAttention(nn.Module):