Update videoaction.py

Zodiark-ch · Jul 3, 2024 · ffd9eca · ffd9eca
1 parent 302b5c1
commit ffd9eca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alldatasets/videoaction.py b/alldatasets/videoaction.py
@@ -123,10 +123,10 @@ def collate_fn(self,batch):
         batch_cls=torch.zeros(batch,max_segment_len,300,2048).cuda()
         for filename in range(len(file)):
             for file_name in range(len(file[filename])):
-                cls=np.load('/home/data/dddd/data4/%s'%(file[filename][file_name]))
+                cls=np.load('/data/causaction/pretrain_representation/%s'%(file[filename][file_name]))
                 batch_cls[filename][file_name]=torch.from_numpy(cls)
 
         return batch_ids,batch_doc_len,batch_label,batch_label_mask,batch_cls,batch_seg_id,batch_action,batch_adj_mask
 
 
-
+