diff --git a/dataset/tiktok_video_dataset.py b/dataset/tiktok_video_dataset.py index 9982b0a..6fbcde1 100644 --- a/dataset/tiktok_video_dataset.py +++ b/dataset/tiktok_video_dataset.py @@ -293,7 +293,7 @@ def __getitem__(self, idx): idx = int(idx * self.train_sample_interval) idx = idx + random.randint(0, self.train_sample_interval - 1) idx = min(idx, len(self) - 1) - elif self.split == "eval": + elif self.split == "val": idx = int(idx * self.eval_sample_interval) raw_data = self.get_metadata(idx)