mshukor commited on
Commit
8cd3695
1 Parent(s): 5c12f0c

Update models/timesformer.py

Browse files
Files changed (1) hide show
  1. models/timesformer.py +3 -0
models/timesformer.py CHANGED
@@ -10,6 +10,9 @@ import warnings
10
  import torch.nn.functional as F
11
  import numpy as np
12
 
 
 
 
13
  from timesformer.models.vit_utils import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
14
  from timesformer.models.helpers import load_pretrained
15
  from timesformer.models.vit_utils import DropPath, to_2tuple, trunc_normal_
 
10
  import torch.nn.functional as F
11
  import numpy as np
12
 
13
+ import os, sys
14
+ sys.path.append("../TimeSformer ")
15
+
16
  from timesformer.models.vit_utils import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
17
  from timesformer.models.helpers import load_pretrained
18
  from timesformer.models.vit_utils import DropPath, to_2tuple, trunc_normal_