boopathiraj commited on
Commit
4f3cffa
·
verified ·
1 Parent(s): d8953e6

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modnet.py +5 -1
modnet.py CHANGED
@@ -2,8 +2,12 @@ import torch
2
  import torch.nn as nn
3
  import torch.nn.functional as F
4
 
5
- from .src.models.backbones import SUPPORTED_BACKBONES
 
6
 
 
 
 
7
 
8
  #------------------------------------------------------------------------------
9
  # MODNet Basic Modules
 
2
  import torch.nn as nn
3
  import torch.nn.functional as F
4
 
5
+ import sys
6
+ import os
7
 
8
+ sys.path.append(os.path.dirname(__file__))
9
+
10
+ from src.models.backbones import SUPPORTED_BACKBONES
11
 
12
  #------------------------------------------------------------------------------
13
  # MODNet Basic Modules