We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 90bccb3 commit c55d393Copy full SHA for c55d393
openunreid/data/datasets/vehicleid.py
@@ -33,6 +33,7 @@ def __init__(
33
self, root, mode, val_split=0.2, test_size=800, del_labels=False, **kwargs
34
):
35
self.root = osp.abspath(osp.expanduser(root))
36
+ self.model = mode
37
self.dataset_dir = osp.join(self.root, self.dataset_dir)
38
self.del_labels = del_labels
39
self.download_dataset(self.dataset_dir, self.dataset_url)
0 commit comments