From 5d27504a06cb6ece98e14095fb2952fd57dab3b5 Mon Sep 17 00:00:00 2001
From: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Date: Thu, 25 Aug 2022 16:53:32 +0800
Subject: [PATCH] [Fix]: fix deep copy data list item (#471)
---
mmengine/dataset/base_dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mmengine/dataset/base_dataset.py b/mmengine/dataset/base_dataset.py
index 3aa56475..fcbb24a5 100644
--- a/mmengine/dataset/base_dataset.py
+++ b/mmengine/dataset/base_dataset.py
@@ -264,7 +264,7 @@ class BaseDataset(Dataset):
self.data_bytes[start_addr:end_addr]) # type: ignore
data_info = pickle.loads(bytes) # type: ignore
else:
- data_info = self.data_list[idx]
+ data_info = copy.deepcopy(self.data_list[idx])
# Some codebase needs `sample_idx` of data information. Here we convert
# the idx to a positive number and save it in data information.
if idx >= 0:
--
GitLab