SupermanxKiaski commited on
Commit
f9ebb4f
1 Parent(s): c9cbb49

Update datasets/video_dataset.py

Browse files
Files changed (1) hide show
  1. datasets/video_dataset.py +3 -0
datasets/video_dataset.py CHANGED
@@ -1,5 +1,8 @@
1
  import random
2
 
 
 
 
3
  import torch
4
  import torch.nn.functional as F
5
  from torch.utils.data import Dataset
 
1
  import random
2
 
3
+ import sys
4
+ sys.path.insert(0, './models')
5
+
6
  import torch
7
  import torch.nn.functional as F
8
  from torch.utils.data import Dataset