Skip to content

Commit c55d393

Browse files
authored
Update vehicleid.py
1 parent 90bccb3 commit c55d393

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

openunreid/data/datasets/vehicleid.py

+1
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ def __init__(
3333
self, root, mode, val_split=0.2, test_size=800, del_labels=False, **kwargs
3434
):
3535
self.root = osp.abspath(osp.expanduser(root))
36+
self.model = mode
3637
self.dataset_dir = osp.join(self.root, self.dataset_dir)
3738
self.del_labels = del_labels
3839
self.download_dataset(self.dataset_dir, self.dataset_url)

0 commit comments

Comments
 (0)