diff --git a/torchreid/data/datasets/dataset.py b/torchreid/data/datasets/dataset.py index bf1db07..6a41a55 100644 --- a/torchreid/data/datasets/dataset.py +++ b/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): diff --git a/torchreid/engine/video/triplet.py b/torchreid/engine/video/triplet.py index 204a583..c1717c7 100644 --- a/torchreid/engine/video/triplet.py +++ b/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,