update code structure

2018-07-02 16:39:06 +01:00 · 2018-07-02 16:39:06 +01:00 · 63ebdf5071
parent d9997b40d6
commit 63ebdf5071
1 changed files with 1 additions and 1 deletions
--- a/dataset_loader.py
+++ b/dataset_loader.py
@ -120,4 +120,4 @@ class VideoDataset(Dataset):
            imgs.append(img)
        imgs = torch.cat(imgs, dim=0)

-        return imgs, pid, camid
+        return imgs, pid, camid