Update models/timesformer.py
Browse files- 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_
|