update

2025-06-04 13:46:52 +08:00
parent cd436605a5
commit c0c4f9b610
13 changed files with 629 additions and 25 deletions
--- a/core/seq_dataset_preprocessed.py
+++ b/core/seq_dataset_preprocessed.py
@@ -21,7 +21,7 @@ class SeqReconstructionDatasetPreprocessed(BaseDataset):
        super(SeqReconstructionDatasetPreprocessed, self).__init__(config)
        self.config = config
        self.root_dir = config["root_dir"]
-        self.real_root_dir = r"/media/hofee/repository/final_test_set/view"
+        self.real_root_dir = r"/media/hofee/data/data/new_testset"
        self.item_list = os.listdir(self.root_dir)

    def __getitem__(self, index):