summaryrefslogtreecommitdiff
path: root/models/model.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-03-15 19:41:28 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-03-15 19:41:58 +0800
commit864fca2c9ca65847c0f1f318dfe50a1e6155e418 (patch)
treecfc2668a81317d7d04a043f2fa1bf0a0087da719 /models/model.py
parentfe6cd66d19c16153322577fb13779020934cf1e2 (diff)
Fix redundant gallery_dataset_meta assignment
Diffstat (limited to 'models/model.py')
-rw-r--r--models/model.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/models/model.py b/models/model.py
index 5966ae1..c350d11 100644
--- a/models/model.py
+++ b/models/model.py
@@ -425,7 +425,6 @@ class Model:
unit='clips'):
gallery_samples_c.append(self._get_eval_sample(sample))
gallery_samples[condition] = default_collate(gallery_samples_c)
- gallery_samples['meta'] = self._gallery_dataset_meta
# Probe
probe_samples_c = []
for sample in tqdm(probe_dataloader,
@@ -435,6 +434,7 @@ class Model:
probe_samples_c = default_collate(probe_samples_c)
probe_samples_c['meta'] = self._probe_datasets_meta[condition]
probe_samples[condition] = probe_samples_c
+ gallery_samples['meta'] = self._gallery_dataset_meta
return gallery_samples, probe_samples