mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

change batch_size in video

pull/133/head
KaiyangZhou 2019-03-21 14:12:50 +00:00
parent 77c47b4e53
commit 75dd835e42
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torchreid/data/datamanager.py
View File

@ -163,7 +163,7 @@ class VideoDataManager(DataManager):
def __init__(self, root, sources, targets=None, height=256, width=128, random_erase=False,
color_jitter=False, color_aug=False, use_cpu=False, split_id=0, combineall=False,
batch_size=32, workers=4, num_instances=4, train_sampler=None,
batch_size=3, workers=4, num_instances=4, train_sampler=None,
seq_len=15, sample_method='evenly'):
super(VideoDataManager, self).__init__(sources, targets=targets, height=height, width=width,