bug fix: pass video-specific args to dataset.__add__

torchreid/data/datasets/dataset.py

@@ -97,7 +97,9 @@ class Dataset(object):
                 transform=self.transform,
                 mode=self.mode,
                 combineall=False,
-                verbose=False
+                verbose=False,
+                seq_len=self.seq_len,
+                sample_method=self.sample_method
             )
 
     def __radd__(self, other):

torchreid/engine/video/triplet.py

@@ -73,7 +73,7 @@ class VideoTripletEngine(ImageTripletEngine, VideoSoftmaxEngine):
     """
 
     def __init__(self, datamanager, model, optimizer, margin=0.3,
-                 weight_t=1, weight_x=1, scheduler=None, use_gpu=False,
+                 weight_t=1, weight_x=1, scheduler=None, use_gpu=True,
                  label_smooth=True, pooling_method='avg'):
         super(VideoTripletEngine, self).__init__(datamanager, model, optimizer, margin=margin,
                                                  weight_t=weight_t, weight_x=weight_x,