From d7ce92975c74ce6bdba635ed1a3bb2766222aba9 Mon Sep 17 00:00:00 2001
From: YWMditto <ditto@YWMdittodeMacBook-Pro.local>
Date: Mon, 8 Aug 2022 19:35:04 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=BA=20=5FTruncatedDataLoader=20=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0=20=5F=5Fsetattr=5F=5F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fastNLP/core/controllers/utils/utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/fastNLP/core/controllers/utils/utils.py b/fastNLP/core/controllers/utils/utils.py
index 0a351354..b01cccc4 100644
--- a/fastNLP/core/controllers/utils/utils.py
+++ b/fastNLP/core/controllers/utils/utils.py
@@ -132,6 +132,10 @@ class _TruncatedDataLoader:
     def __getattr__(self, item):
         return getattr(self.dataloader, item)
 
+    def __setattr__(self, key, value):
+        # 添加该函数使得在进行实验性训练或者评测时，用户对于 trainer.dataloader 的感觉和正常训练完全一样；
+        setattr(self.dataloader, key, value)
+
 
 def check_evaluate_every(evaluate_every):
     r"""